question_paper

question_paper.py

@@ -1,11 +1,7 @@
-import time
-import copy
-import asyncio
-import requests
 import transformers
 import torch
 
-from fastapi import FastAPI, Request
+from fastapi import FastAPI
 
 from transformers import AutoTokenizer
 
@@ -13,6 +9,11 @@ from transformers import AutoTokenizer
 
 app = FastAPI()
 model = "meta-llama/Llama-2-70b"
+
+@app.get("/")
+def read_root():
+    return {"Hello": "World!"}
+
 @app.get("/llama")
 def llama():
     tokenizer = AutoTokenizer.from_pretrained(model)