Hotfix

app.py

@@ -23,7 +23,7 @@ def greet_json():
 @app.post("/generate_chat")
 def generateAi(request: ChatRequest):
     messages = [
-        {"role": "system", "content": "You are a helpful assistant."},
+        {"role": "system", "content": "You are a Mandarin language learning assistant that only answers in Mandarin."},
         {"role": "user", "content": request.prompt}
     ]
     text = tokenizer.apply_chat_template(

requirements.txt

@@ -2,4 +2,6 @@ fastapi
 uvicorn[standard]
 transformers
 pydantic
-accelerator
+accelerator
+torch
+torchvision