added simple gradio

app.py

@@ -1,3 +1,10 @@
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM
+
+tokenizer = AutoTokenizer.from_pretrained("bigscience/T0pp")
+model = AutoModelForSeq2SeqLM.from_pretrained("bigscience/T0pp")
+#tokenizer = AutoTokenizer.from_pretrained("ethzanalytics/ai-msgbot-gpt2-M")
+#model = AutoModelForCausalLM.from_pretrained("ethzanalytics/ai-msgbot-gpt2-M")
+
 import gradio as gr
 import random
 
@@ -10,7 +17,11 @@ def chat(message):
     elif message.startswith("Where"):
         response = random.choice(["Here", "There", "Somewhere"])
     else:
-        response = "I don't know"
+        inputs = tokenizer.encode(message, return_tensors="pt")
+        input_len = len(message)
+        outputs = model.generate(inputs)
+        response = tokenizer.decode(outputs[0])[input_len:]
+
     history.append((message, response))
     gr.set_state(history)
     html = "<div class='chatbot'>"
@@ -27,3 +38,4 @@ iface = gr.Interface(chat, "text", "html", css="""
     .resp_msg {background-color:lightgray;align-self:self-end}
 """, allow_screenshot=False, allow_flagging=False)
 iface.launch()
+