Update app.py

app.py

@@ -1,18 +1,26 @@
 import gradio as gr
 from transformers import pipeline, Conversation
-import torch  # Ensures that PyTorch is available
+import torch
 
-# Load the conversational pipeline with a local model
-chatbot = pipeline("conversational", model="facebook/blenderbot-400M-distill")
+# Load the conversational pipeline with the model
+chatbot = pipeline(
+    "conversational", 
+    model="facebook/blenderbot-400M-distill",
+    device=0 if torch.cuda.is_available() else -1  # Use GPU if available
+)
 
 def chat(user_input):
-    # Initialize a conversation with the user's message
-    conversation = Conversation(user_input)
-    # Process the conversation with the model
-    updated_conversation = chatbot(conversation)
-    # Return the latest generated response
-    return updated_conversation.generated_responses[-1]
+    try:
+        # Initialize a conversation with the user's message
+        conversation = Conversation(user_input)
+        # Process the conversation with the model
+        updated_conversation = chatbot(conversation)
+        # Return the latest generated response
+        return updated_conversation.generated_responses[-1]
+    except Exception as e:
+        return f"An error occurred: {str(e)}"
 
+# Create the Gradio interface
 iface = gr.Interface(
     fn=chat,
     inputs=gr.Textbox(lines=2, placeholder="Type your message here..."),
@@ -22,4 +30,4 @@ iface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()