Update app.py

app.py

@@ -1,22 +1,25 @@
+import openai
 import gradio as gr
-from transformers import pipeline
+openai.api_key ='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'
+def openai_chat(prompt):
+    completions = openai.Completion.create(
+        engine="text-davinci-003",
+        prompt=prompt+"The following is the prompt from teacher working in canvas infrastructure",
+        max_tokens=1024,
+        n=1,
+        temperature=0.5,
+        frequency_penalty=0,
+        presence_penalty=0.6,
+        stop=[" Human:", " AI:"]
+    )
 
-generator = pipeline('text-generation', model='gpt2')
-
-def generate(text):
-    result = generator(text, max_length=30, num_return_sequences=1)
-    return result[0]["generated_text"]
-
-examples = [
-    ["The Moon's orbit around Earth has"],
-    ["The smooth Borealis basin in the Northern Hemisphere covers 40%"],
-]
-
-demo = gr.Interface(
-    fn=generate,
-    inputs=gr.inputs.Textbox(lines=5, label="Input Text"),
-    outputs=gr.outputs.Textbox(label="Generated Text"),
-    examples=examples
-)
-
-demo.launch()
+    message = completions.choices[0].text
+    return message.strip()
+def chatbot(input, history=[]):
+    output = openai_chat(input)
+    history.append((input, output))
+    return history, history
+gr.Interface(fn = chatbot,
+             inputs = ["text",'state'],
+             outputs = ["chatbot",'state'],
+            allow_flagging="manual").launch()