fl399 commited on
Commit
870a50b
·
1 Parent(s): 44a663b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -156,11 +156,11 @@ def evaluate(
156
  # output = tokenizer.decode(s)
157
  output = query({
158
  "inputs": prompt
159
- })
160
  elif llm == "flan-ul2":
161
  output = query({
162
  "inputs": prompt
163
- })
164
 
165
  else:
166
  RuntimeError(f"No such LLM: {llm}")
 
156
  # output = tokenizer.decode(s)
157
  output = query({
158
  "inputs": prompt
159
+ })["generated_text"]
160
  elif llm == "flan-ul2":
161
  output = query({
162
  "inputs": prompt
163
+ })["generated_text"]
164
 
165
  else:
166
  RuntimeError(f"No such LLM: {llm}")