Update app.py

app.py

@@ -1,21 +1,31 @@
+import streamlit as st
 import os
 from langchain import HuggingFaceHub,PromptTemplate,LLMChain
 
 from getpass import getpass
 #HUGGINGFACE_API_TOKEN=getpass()
 os.environ["HUGGINGFACEHUB_API_TOKEN"] = "Token"
+
 model_id="mistralai/Mistral-7B-Instruct-v0.2"
 conv_model=HuggingFaceHub(huggingfacehub_api_token=os.environ['HUGGINGFACEHUB_API_TOKEN'],
                           repo_id=model_id,
-                          model_kwargs={"temperature":0.5,"max_new_tokens":5000})
+                          model_kwargs={"temperature":0.2,"max_new_tokens":5000})
 
 def answer_question(question):
   prompt = PromptTemplate( input_variables=['question'],
-                          template="""You are a logical and a numerical question solver.Could you please answer this question ?{question}
+                          template="""You are a logical and a numerical question solver.Could you please answer this question ? {question}
+                          Show all the steps in details to solve the problem. 
                           """)
   conv_chain = LLMChain(llm=conv_model, prompt=prompt, verbose=True,output_key="Answer")
   response = conv_chain({'question':question})
   return response
+
+st.title("Zero Shot Question Answering")
+input=st.text_input("Input: ",key="input")
+submit=st.button("Ask the question")
+if submit and input:
+    response=answer_question(input)
+    st.write(response['Answer'])
   
 if __name__ == "__main__":
   print("nothing given")