Update app.py

app.py

@@ -1,5 +1,5 @@
-import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
 
 # Load pre-trained GPT-2 model and tokenizer
 model_name = "google/gemma-7b"
@@ -19,11 +19,14 @@ def generate_response(prompt):
 
     return response
 
-gr.Interface(
-    fn=generate_response,
-    inputs=gr.Textbox(lines=2, placeholder="Enter your message here..."),
-    outputs="text",
-    title="Akhil's Chatbot (powered by GPT2)",
-    description="This is a Basuc chatbot using OpenAI's GPT2. Ask it anything!",
-    theme="compact"
-).launch()
+# Spaces-compatible function
+def spaces_chatbot(input_dict):
+    prompt = input_dict["text"]
+    response = generate_response(prompt)
+    return {"response": response}
+
+# Sample input
+sample_input = {"text": "Hello, how are you?"}
+
+# Test the function
+print(spaces_chatbot(sample_input))