Update app.py

app.py

@@ -1,12 +1,11 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, Request, HTTPException
 from fastapi.responses import JSONResponse
 from transformers import pipeline
-from fastapi.requests import Request
 
 app = FastAPI()
 
-# Initialize the text generation pipeline
-pipe = pipeline("text-generation", model="microsoft/Phi-3-mini-4k-instruct")
+# Initialize the text generation pipeline with trust_remote_code=True
+pipe = pipeline("text-generation", model="microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True)
 
 @app.middleware("http")
 async def log_requests(request: Request, call_next):
@@ -32,4 +31,4 @@ def generate(payload: dict):
         return JSONResponse(
             status_code=500,
             content={"error": f"An error occurred: {str(e)}"}
-        )
+        )