Update app.py

app.py

@@ -6,28 +6,41 @@ import time
 
 # Default model
 MODEL = "ZeppelinCorp/Charm_15"
-text_generator = pipeline("text-generation", model=MODEL)
+text_generator = None
 
 def load_model(model_name):
-    return pipeline("text-generation", model=model_name)
+    try:
+        return pipeline("text-generation", model=model_name)
+    except Exception as e:
+        return None
 
 def generate_response(prompt, model_name):
     global MODEL, text_generator
-    if model_name != MODEL:
+    
+    if model_name != MODEL or text_generator is None:
         MODEL = model_name
         text_generator = load_model(model_name)
+        if text_generator is None:
+            return "Error loading model. Please try another one."
     
     loading_animation = gr.update(value=". . .", visible=True)
     time.sleep(1)  # Simulate loading delay
-    response = text_generator(prompt, max_length=200)[0]['generated_text']
     
-    return gr.update(value=response, visible=True)
+    try:
+        response = text_generator(prompt, max_length=200)[0]['generated_text']
+        return gr.update(value=response, visible=True)
+    except Exception as e:
+        return f"Error generating response: {e}"
 
 def google_search(query):
     loading_animation = gr.update(value="Searching…", visible=True)
     time.sleep(1)  # Simulate search delay
-    results = [url for url in search(query, num=5, stop=5, pause=2)]
-    return "\n".join(results)
+    
+    try:
+        results = [url for url in search(query, num=5, stop=5, pause=2)]
+        return "\n".join(results)
+    except Exception as e:
+        return f"Error in search: {e}"
 
 def upload_image(image):
     return "Image uploaded successfully."
@@ -40,8 +53,12 @@ def analyze_reasoning(input_text):
 
 def switch_model(new_model):
     global MODEL, text_generator
-    MODEL = new_model
+    
     text_generator = load_model(new_model)
+    if text_generator is None:
+        return f"Failed to switch to {new_model}. Try another model."
+    
+    MODEL = new_model
     return f"Switched to {new_model}"
 
 iface = gr.Interface(