Spaces:
Runtime error
Runtime error
Yash Sachdeva
commited on
Commit
•
d618858
1
Parent(s):
3dd4624
question
Browse files- question_paper.py +1 -1
question_paper.py
CHANGED
|
@@ -32,7 +32,7 @@ def llama(input):
|
|
| 32 |
# print(tresponse)
|
| 33 |
output = generator(input, max_length=100)
|
| 34 |
|
| 35 |
-
response_message = {"message": output["generated_text"]}
|
| 36 |
json_response = jsonable_encoder(response_message)
|
| 37 |
return JSONResponse(content=json_response)
|
| 38 |
|
|
|
|
| 32 |
# print(tresponse)
|
| 33 |
output = generator(input, max_length=100)
|
| 34 |
|
| 35 |
+
response_message = {"message": output[0]["generated_text"]}
|
| 36 |
json_response = jsonable_encoder(response_message)
|
| 37 |
return JSONResponse(content=json_response)
|
| 38 |
|