Update app.py

app.py

@@ -4,12 +4,31 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
+openai.api_key = os.getenv("OPENAI_API_KEY")
+
+    def predict(input, history=[]):
+        
+        new_user_input_ids = input
+        response = openai.Completion.create(
+        model="davinci:ft-placeholder-2022-12-10-04-13-26",
+        prompt=input
+        temperature=0.13,
+        max_tokens=310,
+        top_p=1,
+        frequency_penalty=0.36,
+        presence_penalty=1.25
+        )
+
+    # generate a response 
+        history = model.generate(bot_input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id).tolist()
+    
+    # convert the tokens to text, and then split the responses into lines
+        response = tokenizer.decode(history[0]).split("<|endoftext|>")
+        response = [(response[i], response[i+1]) for i in range(0, len(response)-1, 2)]  # convert to tuples of list
+        return response, history
+
 
-openai.organization = "org-orRhfBkKOfOuNACbjPyWKbUt"
-openai.api_key = "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"
 
-def predict(input, history=[]):
 
     new_user_input_ids = tokenizer.encode(input + tokenizer.eos_token, return_tensors='pt')
     
@@ -27,28 +46,6 @@ def predict(input, history=[]):
     frequency_penalty=0,
     presence_penalty=0).tolist()
 
-    # write some HTML
-    html = "<div class='chatbot'>"
-    for m, msg in enumerate(response):
-        cls = "user" if m%2 == 0 else "bot"
-        html += "<div class='msg {}'> {}</div>".format(cls, msg)
-    html += "</div>"
-
-    history = response[Completion]
-    
-    # convert the tokens to text, and then split the responses into lines
-    response = tokenizer.decode(history[0]).split
-
-css = """
-.chatbox {display:flex;flex-direction:row}
-.msg {padding:4px;margin-bottom:4px;border-radius:4px;width:80%}
-.msg.user {background-color:cornflowerblue;color:white}
-.msg.bot {background-color:lightgray;align-self:self-end}
-.footer {display:none !important}
-"""
-
 gr.Interface(fn=predict,
-             theme="default",
-             inputs=[gr.inputs.Textbox(placeholder="I'm AI-DHD - ask me anything!"), "state"],
-             outputs=["html", "state"],
-             css=css).launch()
+             inputs=["text", "state"],
+             outputs=["chatbot", "state"]).launch()