Update app/main.py

app/main.py

@@ -8,8 +8,8 @@ app = FastAPI()
 
 # دانلود مدل و توکنایزر
 model_name = "bert-base-uncased"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name, force_download=True)
+model = AutoModelForCausalLM.from_pretrained(model_name, is_decoder=True, force_download=True)
 
 # انتقال مدل به CPU
 model.to("cpu")
@@ -21,7 +21,7 @@ class GenerateRequest(BaseModel):
     prompt: str
     image: str = ""
 
-@app.post('/generate')
+@app.post('/')
 async def generate(request: GenerateRequest):
     user_input = request.prompt
     image_path = request.image