Update app.py

app.py

@@ -1,25 +1,30 @@
-from fastapi import FastAPI, Request
+from fastapi import FastAPI, Request, HTTPException
 from transformers import pipeline
 import os
 
 app = FastAPI()
 
-# Встановлення директорії кешу
 cache_dir = os.path.expanduser("~/.cache/huggingface/transformers")
 os.environ["TRANSFORMERS_CACHE"] = cache_dir
 
-pipe = pipeline(
-    "text-generation",
-    model="google/gemma-3-1b-it",
-    token=os.environ["HF_TOKEN"],
-    device="cpu",
-    max_new_tokens=256,
-    temperature=0.7
-)
+try:
+    pipe = pipeline(
+        "text-generation",
+        model="google/gemma-3-1b-it",
+        token=os.environ["HF_TOKEN"],
+        device="cpu",
+        max_new_tokens=256,
+        temperature=0.7
+    )
+except Exception as e:
+    raise HTTPException(status_code=500, detail=str(e))
 
 @app.post("/generate")
 async def generate(request: Request):
     data = await request.json()
     prompt = data.get("prompt", "")
-    result = pipe(prompt)
-    return {"response": result[0]["generated_text"]}
+    try:
+        result = pipe(prompt)
+        return {"response": result[0]["generated_text"]}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))