Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import gradio as gr
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
+import torch
 
 # Clone the Gradio repository
 os.system("git clone https://github.com/gradio-app/gradio.git")
@@ -11,6 +12,9 @@ os.system("cd gradio && pip install .")
 # Install transformers
 os.system("pip install transformers")
 
+# Install PyTorch
+os.system("pip install torch==1.10.0+cpu torchvision==0.11.1+cpu torchaudio==0.10.0+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html")
+
 class ArabicAssistant:
     def __init__(self):
         self.model_name = "EleutherAI/gpt-neo-2.7B"
@@ -40,5 +44,5 @@ def chatbot_interaction(user_input):
 
     return response
 
-iface = gr.Interface(fn=chatbot_interaction, inputs="text", outputs="text", live=True, theme="huggingface")
+iface = gr.Interface(fn=chatbot_interaction, inputs=gr.Textbox(), outputs=gr.Textbox(), live=True, theme="huggingface", layout="vertical", title="Arabic Assistant Chatbot", description="Chat with the Arabic Assistant in the Egyptian dialect.")
 iface.launch()