fl399 commited on
Commit
a82c90c
1 Parent(s): 970f64b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -186,10 +186,9 @@ def evaluate(
186
  output = tokenizer.decode(s)
187
  elif llm == "flan-ul2":
188
  try:
189
- res = query({"inputs": prompt_0shot})
190
  except:
191
  output = "<flan-ul2 inference API error - try later>"
192
- output = res[0]["generated_text"]
193
  elif llm == "gpt-3.5-turbo":
194
  try:
195
  output = get_response_from_openai(prompt_0shot)
 
186
  output = tokenizer.decode(s)
187
  elif llm == "flan-ul2":
188
  try:
189
+ output = query({"inputs": prompt_0shot})[0]["generated_text"]
190
  except:
191
  output = "<flan-ul2 inference API error - try later>"
 
192
  elif llm == "gpt-3.5-turbo":
193
  try:
194
  output = get_response_from_openai(prompt_0shot)