create app.py

First version of the chat interface.

app.py

@@ -0,0 +1,26 @@
+import gradio as gr
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+# Load the pre-trained LLM model and tokenizer
+model_name = "microsoft/DialoGPT-medium"
+model = AutoModelForCausalLM.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+# Define the function to generate the chatbot response
+def chatbot(input_text):
+    input_ids = tokenizer.encode(input_text, return_tensors="pt")
+    output = model.generate(input_ids, max_length=1000, do_sample=True, top_p=0.92, top_k=0, num_return_sequences=1)
+    response = tokenizer.decode(output[0], skip_special_tokens=True)
+    return response
+
+# Create the Gradio interface
+chat_interface = gr.Blocks()
+
+with chat_interface:
+    gr.Markdown("# Chatbot")
+    chatbot_input = gr.Textbox(placeholder="Type your message here...")
+    chatbot_output = gr.Textbox(label="Chatbot Response")
+    chat_btn = gr.Button("Send")
+    chat_btn.click(chatbot, inputs=chatbot_input, outputs=chatbot_output)
+
+chat_interface.launch()