Update app.py

app.py

@@ -1,7 +1,20 @@
 import gradio as gr
 
-def greet(name):
-    return "Hello " + name + "!!"
+def chat_with_phi2(input_text):
+    # Encoding the input text
+    encoded_input = tokenizer(input_text, return_tensors='pt')
+    # Generating output using the fine-tuned model
+    output = model.generate(**encoded_input, max_length=50)
+    # Decoding the output tokens to a string
+    response = tokenizer.decode(output[0], skip_special_tokens=True)
+    return response
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+interface = gr.Interface(
+    fn=chat_with_phi2,
+    inputs=gr.inputs.Textbox(lines=2, placeholder="Ask something about W&L..."),
+    outputs="text",
+    title="W&L ChatGPT-like AI",
+    description="Ask any question about Washington and Lee University!"
+)
+
+interface.launch()