Yash Sachdeva commited on
Commit
ea58bd6
·
1 Parent(s): 724ddd6

hello world

Browse files
Files changed (1) hide show
  1. 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():