Update app.py

app.py

@@ -4,16 +4,14 @@ import gradio as gr
 model = GPTNeoForCausalLM.from_pretrained("EleutherAI/gpt-neo-125M")
 tokenizer = GPT2Tokenizer.from_pretrained("EleutherAI/gpt-neo-125M")
 
-prompt = """This is a discussion between a person and an entrepreneur. 
+prompt = """This is a discussion between a person and Hassan Kane, an entrepreneur. 
 
-person: What is your name?
-entrepreneur: Hassan
-person: Where are you working?
-entrepreneur: It's like one of these fancy adjustable height desk
-person: What will you work on?
-entrepreneur: The international development hackathon
 person: What are you working on?
-entrepreneur: Developping an iPhone app
+Hassan: This new AI community building the future of Africa
+person: Where are you?
+Hassan: In Lagos for a week, then Paris or London.
+person: How's it going? 
+Hassan: Not bad.. Just trying to hit EV (escape velocity) with my startup
 person: """
 
 def my_split(s, seps):
@@ -39,7 +37,7 @@ def chat_base(input):
 # print(">>>", result)
   return result
   
-  import gradio as gr
+import gradio as gr
 
 def chat(message):
     history = gr.get_state() or []
@@ -52,12 +50,7 @@ def chat(message):
         html += f"<div class='user_msg'>{user_msg}</div>"
         html += f"<div class='resp_msg'>{resp_msg}</div>"
     html += "</div>"
-    return html
+    return response
 
-iface = gr.Interface(chat, gr.inputs.Textbox(label="Ask Hassan a Question"), "html", css="""
-    .chatbox {display:flex;flex-direction:column}
-    .user_msg, .resp_msg {padding:4px;margin-bottom:4px;border-radius:4px;width:80%}
-    .user_msg {background-color:cornflowerblue;color:white;align-self:start}
-    .resp_msg {background-color:lightgray;align-self:self-end}
-""", allow_screenshot=False, allow_flagging=False)
+iface = gr.Interface(chat_base, gr.inputs.Textbox(label="Ask Hassan a Question"), "text", allow_screenshot=False, allow_flagging=False)
 iface.launch()