Update test_gradio.py

test_gradio.py

@@ -3,8 +3,7 @@ from langchain.prompts import PromptTemplate
 from langchain.chains import LLMChain
 from langchain.llms import GooglePalm
 
-
-llm = GooglePalm(temperature=0.1, google_api_key= "AIzaSyB5XIeNPyhIy29g4DUNZzVBKfsa-fVZtrk")
+llm = GooglePalm(temperature=0.1, google_api_key="AIzaSyB5XIeNPyhIy29g4DUNZzVBKfsa-fVZtrk")
 
 template = """Question: {question}
 
@@ -12,7 +11,7 @@ Answer: Let's think step by step."""
 
 prompt_open = PromptTemplate(template=template, input_variables=["question"])
 
-open_chain = LLMChain(prompt=prompt_open,llm = llm)
+open_chain = LLMChain(prompt=prompt_open, llm=llm)
 
 
 def predict(message, history):
@@ -20,10 +19,10 @@ def predict(message, history):
     return gpt_response
 
 gr.Markdown(
-"""
+    """
 # Let's think step by step!
 I will break your question into steps :)
+"""
+)
 
-""")
-    
 gr.ChatInterface(predict).launch(share=False)