Update app.py

app.py

@@ -26,10 +26,7 @@ def load_model():
     pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
     return pipe
 
-
 def main():
-
-    # st.set_page_config(page_title="ChatGPT Clone", page_icon="🤖")
     st.title("ChatGPT-Clone")
 
     # Load the generator model only once
@@ -43,33 +40,27 @@ def main():
 
     # Display past messages
     for msg in st.session_state.messages:
-            with st.chat_message(msg["role"]):
-                st.markdown(msg["content"])
+        with st.chat_message(msg["role"]):
+            st.markdown(msg["content"])
 
     # Chat input
     if prompt := st.chat_input("Ask anything..."):
-        st.session_state.messages.append({"role" : "user", "content" : prompt})
+        st.session_state.messages.append({"role": "user", "content": prompt})
 
         with st.chat_message("user"):
             st.markdown(prompt)
 
         with st.chat_message("assistant"):
             with st.spinner("Thinking..."):
-                # Call Mistral-7B API
                 result = st.session_state.generator(
                     prompt,
-                    max_new_tokens = 512,
-                    temperature = 0.7,
-                    do_sample = True,
-                     
+                    max_new_tokens=512,
+                    temperature=0.7,
+                    do_sample=True
                 )[0]["generated_text"]
                 st.markdown(result)
 
-        st.session_state.messages.append({"role" : "assistant", "content" : result})
-
-            
+        st.session_state.messages.append({"role": "assistant", "content": result})
 
 if __name__ == "__main__":
     main()
-
-