Update app.py

app.py

@@ -1,29 +1,8 @@
 import gradio as gr
-from transformers import pipeline
 
-# Define the models you want to include in the dropdown
-models = {
-    "GPT-2": "gpt2",
-    "DistilGPT-2": "distilgpt2",
-    "BERT Base Uncased": "bert-base-uncased",
-    "T5 Small": "t5-small"
-}
+def greet(name):
+    return "Hello " + name + "!"
 
-# Function to generate the response
-def generate_response(model_name, user_input):
-    generator = pipeline(task="text-generation", model=model_name)
-    output = generator(user_input, max_length=50)[0]["generated_text"]
-    return output
+demo = gr.Interface(fn=greet, inputs="text", outputs="text")
 
-# Create the Gradio interface
-interface = gr.Interface(
-    fn=generate_response,
-    inputs=[gr.inputs.Dropdown(list(models.keys()), label="Select Model"),
-            gr.inputs.Textbox(label="Enter your text")],
-    outputs=gr.outputs.Textbox(label="Model Output"),
-    title="Chat with LLMs",
-    description="Select a model and enter a prompt to chat with different large language models."
-)
-
-# Launch the interface
-interface.launch()
+demo.launch()