nestole commited on
Commit
8ffb9c9
·
verified ·
1 Parent(s): f242578

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +5 -1
run.py CHANGED
@@ -136,9 +136,11 @@ def responsecritical(
136
  formatted_promptc = format_prompt(systemc+"\n"+prompt)
137
  streamc = client.text_generation(formatted_promptc, **generate_kwargs, stream=True, details=True, return_full_text=False)
138
  outputc = ""
139
- print(streamc)
140
  for responsec in streamc:
141
  outputc += responsec.token.text
 
 
142
  sentence_lower = outputc.lower()
143
  print("Done critcial")
144
  # Check if the word 'nein' is in the sentence
@@ -190,6 +192,8 @@ def response(
190
  output = ""
191
  for response in stream:
192
  output += response.token.text
 
 
193
  osaanswer=simosa(prompt, osalist, osa)
194
  output=output[:-4]+"\n"+osaanswer
195
  yield output
 
136
  formatted_promptc = format_prompt(systemc+"\n"+prompt)
137
  streamc = client.text_generation(formatted_promptc, **generate_kwargs, stream=True, details=True, return_full_text=False)
138
  outputc = ""
139
+ #print(streamc)
140
  for responsec in streamc:
141
  outputc += responsec.token.text
142
+ print("CRITICAL")
143
+ print(outputc)
144
  sentence_lower = outputc.lower()
145
  print("Done critcial")
146
  # Check if the word 'nein' is in the sentence
 
192
  output = ""
193
  for response in stream:
194
  output += response.token.text
195
+ print("FINAL")
196
+ print(output)
197
  osaanswer=simosa(prompt, osalist, osa)
198
  output=output[:-4]+"\n"+osaanswer
199
  yield output