Update app.py

app.py

@@ -1,3 +1,22 @@
 import gradio as gr
+from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
 
-gr.load("models/IEEEVITPune-AI-Team/ChatbotAlpha0.7").launch()
+# Load the model and tokenizer
+model_name = "IEEEVITPune-AI-Team/ChatbotAlpha0.7"
+model = AutoModelForCausalLM.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+# Define the function to generate response
+def generate_response(prompt):
+    instruction = f"### Instruction:\n{prompt}\n\n### Response:\n"
+    pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=256)
+    result = pipe(instruction)
+    generated_text = result[0]['generated_text'][len(instruction):].strip()
+    return generated_text
+
+# Create a Gradio interface
+input_text = gr.inputs.Textbox(lines=3, label="Enter your prompt")
+output_text = gr.outputs.Textbox(label="Response")
+
+gr.Interface(generate_response, inputs=input_text, outputs=output_text, title="Chatbot", description="What is IEEE?.").launch()
+ad("models/IEEEVITPune-AI-Team/ChatbotAlpha0.7").launch()