Rename 2files_app.py to app.py

2files_app.py → app.py

@@ -19,6 +19,7 @@ login(token=HF_HUB_TOKEN)
 hebrew_generator = pipeline("text-generation", model="Norod78/hebrew-gpt_neo-small")
 english_generator = pipeline("text-generation", model="distilgpt2")
 
+
 # Function to detect language
 def detect_language(user_input):
     try:
@@ -27,15 +28,17 @@ def detect_language(user_input):
     except:
         return "unsupported"
 
+
 # Function to generate a response
 def generate_response(text):
     language = detect_language(text)
     if language == "hebrew":
-        return hebrew_generator(text, max_length=100)[0]["generated_text"]
+        return hebrew_generator(text, max_length=100, truncation=True)[0]["generated_text"]
     elif language == "english":
-        return english_generator(text, max_length=100)[0]["generated_text"]
+        return english_generator(text, max_length=100, truncation=True)[0]["generated_text"]
     return "Sorry, I only support Hebrew and English."
 
+
 # FastAPI lifespan event
 @asynccontextmanager
 async def lifespan(app: FastAPI):
@@ -43,13 +46,16 @@ async def lifespan(app: FastAPI):
     yield  # Wait until app closes
     print("Shutting down application...")
 
+
 # Create FastAPI app
 app = FastAPI(lifespan=lifespan)
 
+
 @app.get("/")
 async def root():
     return {"message": "Decision Helper API is running!"}
 
+
 @app.post("/generate_response")
 async def generate_text(request: Request):
     data = await request.json()
@@ -60,6 +66,7 @@ async def generate_text(request: Request):
     response = generate_response(text)
     return {"response": response}
 
+
 # Run the server
 if __name__ == "__main__":
     import uvicorn