Update app.py

app.py

@@ -6,29 +6,16 @@ from langchain.memory import ConversationBufferMemory
 
 OPENAI_API_KEY=os.getenv('OPENAI_API_KEY')
 
-template = """You are a helpful assistant to answer all user queries.
-{chat_history}
-User: {user_message}
-Chatbot:"""
-
-prompt = PromptTemplate(
-    input_variables=["chat_history", "user_message"], template=template
+auth_token = os.getenv("auth_token")
+
+# automatically load the interface from a HF model
+# you can remove the api_key parameter if you don't care about rate limiting.
+demo = gr.load(
+    "huggingface/facebook/wav2vec2-base-960h",
+    title="Speech-to-text",
+    inputs="mic",
+    description="Let me try to guess what you're saying!",
+    hf_token=auth_token
 )
 
-memory = ConversationBufferMemory(memory_key="chat_history")
-
-llm_chain = LLMChain(
-    llm=ChatOpenAI(temperature='0.5', model_name="gpt-3.5-turbo"),
-    prompt=prompt,
-    verbose=True,
-    memory=memory,
-)
-
-def get_text_response(user_message,history):
-    response = llm_chain.predict(user_message = user_message)
-    return response
-
-demo = gr.ChatInterface(get_text_response)
 
-if __name__ == "__main__":
-    demo.launch() #To create a public link, set `share=True` in `launch()`. To enable errors and logs, set `debug=True` in `launch()`.