Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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}")
|