Update app.py

app.py

@@ -2,6 +2,7 @@ from fastapi import FastAPI
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 import os
+from pydantic import BaseModel
 
 # ✅ Force Hugging Face cache to /tmp (writable in Spaces)
 os.environ["HF_HOME"] = "/tmp"