Update app.py

app.py

@@ -5,6 +5,7 @@ from huggingface_hub import InferenceClient
 import gradio as gr
 from pydantic import BaseModel, Field
 from typing import Optional, Literal
+from huggingface_hub.utils._errors import HfHubHTTPError
 
 class PromptInput(BaseModel):
     text: str = Field(..., description="The initial prompt text")
@@ -19,7 +20,7 @@ class RefinementOutput(BaseModel):
 
 class PromptRefiner:
     def __init__(self, api_token: str):
-        self.client = InferenceClient(token=api_token)
+        self.client = InferenceClient(token=api_token,timeout=300)
 
     def refine_prompt(self, prompt_input: PromptInput) -> RefinementOutput:
         if prompt_input.meta_prompt_choice == "morphosis":
@@ -55,12 +56,18 @@ class PromptRefiner:
             response_content = response.choices[0].message.content.strip()
         except HfHubHTTPError as e:
             return (
-                "Error: Model timeout. Please try a smaller model or try again later.",
+                "Error: Model timeout. Please again later.",
                 "",
                 "The selected model is currently experiencing high traffic.",
                 {}
             )
-        
+        except Exception as e:
+            return (
+                f"Error: {str(e)}",
+                "",
+                "An unexpected error occurred.",
+                {}
+            )
         try:
             json_match = re.search(r'<json>\s*(.*?)\s*</json>', response_content, re.DOTALL)
             if json_match: