Update app.py

app.py

@@ -45,13 +45,22 @@ class PromptRefiner:
             {"role": "system", "content": 'You are an expert at refining and extending prompts. Given a basic prompt, provide a more detailed.'},
             {"role": "user", "content": selected_meta_prompt.replace("[Insert initial prompt here]", prompt_input.text)}
         ]
-        response = self.client.chat_completion(
-            model=prompt_refiner_model,
-            messages=messages,
-            max_tokens=2000,
-            temperature=0.8
-        )
-        response_content = response.choices[0].message.content.strip()
+        try:
+            response = self.client.chat_completion(
+                model=prompt_refiner_model,
+                messages=messages,
+                max_tokens=2000,
+                temperature=0.8
+            )
+            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.",
+                "",
+                "The selected model is currently experiencing high traffic.",
+                {}
+            )
+        
         try:
             json_match = re.search(r'<json>\s*(.*?)\s*</json>', response_content, re.DOTALL)
             if json_match: