Update app.py

app.py

@@ -29,10 +29,11 @@ app.add_middleware(
     allow_headers=["*"],  # Allows all headers
 )
 
-# Load NLP models
+humanize_pipe = None  # Default to None before trying to load
+
 try:
-    # Load text humanization model
     humanize_pipe = pipeline("text2text-generation", model="danibor/flan-t5-base-humanizer")
+    print("Humanization model loaded successfully!")
     
     # Load spaCy model
     nlp = spacy.load("en_core_web_sm")
@@ -42,7 +43,7 @@ try:
     
     print("All NLP models loaded successfully!")
 except Exception as e:
-    print(f"Error loading models: {e}")
+    print(f"Error loading humanization model: {e}")
     # Create fallback functions if models fail to load
     def mock_function(text):
         return "Model could not be loaded. This is a fallback response."
@@ -74,6 +75,9 @@ async def humanize_text(request: TextRequest):
     
     try:
         # Generate humanized text
+        if humanize_pipe is None:
+            raise HTTPException(status_code=500, detail="Humanization model is not loaded.")
+
         result = humanize_pipe(input_text, max_length=500, do_sample=True)
         humanized_text = result[0]['generated_text']