Spaces:
Runtime error
Runtime error
Yash Sachdeva
commited on
Commit
·
ea58bd6
1
Parent(s):
724ddd6
hello world
Browse files- question_paper.py +2 -2
question_paper.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import transformers
|
2 |
import torch
|
3 |
|
4 |
-
from fastapi import FastAPI
|
5 |
|
6 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
7 |
|
@@ -20,7 +20,7 @@ def llama(input):
|
|
20 |
# tresponse = TOKENIZER.decode(tokens[0], skip_special_tokens=False)
|
21 |
# print(tresponse)
|
22 |
|
23 |
-
return "hello world"
|
24 |
|
25 |
# @app.on_event("startup")
|
26 |
# def init_model():
|
|
|
1 |
import transformers
|
2 |
import torch
|
3 |
|
4 |
+
from fastapi import FastAPI, Response
|
5 |
|
6 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
7 |
|
|
|
20 |
# tresponse = TOKENIZER.decode(tokens[0], skip_special_tokens=False)
|
21 |
# print(tresponse)
|
22 |
|
23 |
+
return Response(content="hello world", media_type="application/json")
|
24 |
|
25 |
# @app.on_event("startup")
|
26 |
# def init_model():
|