Update app.py

app.py

@@ -4,6 +4,12 @@ from googlesearch import search
 import requests
 import time
 
+# Ensure transformers library is installed
+try:
+    import transformers
+except ImportError:
+    raise ImportError("The 'transformers' library is not installed. Please install it using 'pip install transformers'.")
+
 # Default model
 MODEL = "ZeppelinCorp/Charm_15"
 text_generator = None
@@ -23,19 +29,9 @@ def generate_response(prompt, 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
-    
-    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}"
+    return text_generator(prompt, max_length=200)[0]['generated_text']
 
 def google_search(query):
-    loading_animation = gr.update(value="Searching…", visible=True)
-    time.sleep(1)  # Simulate search delay
-    
     try:
         results = [url for url in search(query, num=5, stop=5, pause=2)]
         return "\n".join(results)
@@ -79,4 +75,4 @@ demo = gr.TabbedInterface(
     ["Chat", "Google Search", "Upload Image", "Upload Audio", "Reasoning", "Switch Model"]
 )
 
-demo.launch()
+demo.launch()