Update app.py

app.py

@@ -1,13 +1,7 @@
 import gradio as gr
 import os
 
-# Check if transformers and torch are installed
-try:
-    import transformers
-    import torch
-except ImportError:
-    # Install transformers and torch if not installed
-    os.system('pip install transformers torch')
+#os.system('pip install transformers torch')
 
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 
@@ -32,11 +26,26 @@ def generate_response(prompt, max_length=50, temperature=0.8):
     
     return response
 
+def chat_room_ui():
+    user_input = gr.Textbox(prompt="User Input", placeholder="Type your message...")
+    chat_display = gr.Textbox(text=chat_history, label="Chat Room", readonly=True)
+    generate_button = gr.Button(text="Generate Response")
+
+    def generate_response_callback():
+        global chat_history
+        user_prompt = user_input.value
+        response = generate_response(user_prompt)
+        chat_history += f" User: {user_prompt} Assistant: {response}"
+        chat_display.value = chat_history
+
+    generate_button.click(generate_response_callback)
+
+    return gr.Column([user_input, chat_display, generate_button])
+
 iface = gr.Interface(
-    fn=generate_response,
-    inputs=gr.Textbox(),
-    outputs="text",
-    live=True
+    fn=None,  # Set fn to None since it will be handled by the UI
+    live=False,
+    ui=chat_room_ui
 )
 
 iface.launch()