Update app.py

app.py

@@ -1,21 +1,13 @@
 import streamlit as st
 from streamlit_chat import message
 import requests
-# from transformers import AutoModelWithLMHead, AutoTokenizer
-
-#tokenizer = AutoTokenizer.from_pretrained('microsoft/DialoGPT-small')
-#model = AutoModelWithLMHead.from_pretrained('output-small-save')
 
 st.set_page_config(
     page_title="SAI",
     page_icon=":robot:"
 )
 
-#API_URL = "https://api-inference.huggingface.co/models/microsoft/DialoGPT-small"
-#headers = {"Authorization": st.secrets['api_key']}
-
 st.header("Hello - Welcome to SAI")
-#st.markdown("[Github](https://github.com/rushic24/DialoGPT-Finetune)")
 st.write("""The Phantom Thief Kid sends a heist notice, warning of another heist. The police deduce that his next target is a recently discovered Fabergé egg, which Suzuki Modern Art Museum in Osaka will display on August 22. The night of the heist, Kid steals the egg and flies off, and Conan and Heiji give chase. However, in the middle of the chase, an unknown assailant shoots Kid in the right eye, and Kid apparently falls into the sea to his death. After recovering the egg, the police fruitlessly search for Kid's body.
 
 The next day, Conan, Ran, and Kogoro board a boat to Tokyo. They meet Natsumi Kousaka, whose great-grandfather worked in Fabergé's factory. She shows them a part of a sketch of two eggs and a key, which were found among her late grandmother's mementos. Conan suspects that the person who shot Kid is on the ship. That night, Ryu Sagawa, a freelance photographer covering the press with news of the egg, is murdered, shot in the right eye in the same fashion as Kid. Soon after his body is discovered, Inspector Megure, along with officers Takagi and Shiratori, arrive by helicopter to inspect the crime scene. At first, they suspect Sonoko's father's servant, Mr. Nishino, but the police and Conan conclude the culprit is Scorpion - a mysterious killer who always shoots his victims in the right eye. A missing lifeboat hints that Scorpion has escaped, and the boat's passengers go to Yokosuka Castle, which holds Scorpion's next target: the second egg.
@@ -29,43 +21,16 @@ if 'generated' not in st.session_state:
 if 'past' not in st.session_state:
     st.session_state['past'] = []
 
-# def query(payload):
-#     bot_input_ids = tokenizer.encode(payload["inputs"]["text"] + tokenizer.eos_token, return_tensors='pt')
-
-#     chat_history_ids = model.generate(
-#       bot_input_ids, max_length=100,
-#       pad_token_id=tokenizer.eos_token_id,  
-#       no_repeat_ngram_size=3,       
-#       do_sample=True, 
-#       top_k=10, 
-#       top_p=0.7,
-#       temperature = 0.8
-#     )
-#     output = tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True)
-#     return {"generated_text": output}
-
 def get_text():
     input_text = st.text_input("You: "," ", key="input")
     return input_text 
 
 
-
-# if user_input:
-    # output = query({
-    #     "inputs": {
-    #         "past_user_inputs": st.session_state.past,
-    #         "generated_responses": st.session_state.generated,
-    #         "text": user_input,
-    #     },"parameters": {"repetition_penalty": 1.33},
-    # })
-    # st.session_state.past.append(user_input)
-    # st.session_state.generated.append(output["generated_text"])
-
 if st.session_state['generated']:
 
     for i in range(len(st.session_state['generated'])-1, -1, -1):
         message(st.session_state["generated"][i], key=str(i))
         message(st.session_state['past'][i], is_user=True, key=str(i) + '_user')
-# message("")
+
 message("Tell me what happened?",is_user=True)
 user_input = get_text()