Update app.py

app.py

@@ -90,14 +90,20 @@ def transcribe_video(youtube_url: str, path: str) -> List[Document]:
     result = client.predict(youtube_url, "translate", True, fn_index=7)
     return [Document(page_content=result[1], metadata=dict(page=1))]
 
-def predict(
-    message: str, system_prompt: str = "", model_url: str = models["Llama2-70b"]["chat_link"]
-) -> Any:
+def predict(message: str, system_prompt: str = '', temperature: float = 0.7, max_new_tokens: int = 1024,
+            topp: float = 0.5, repetition_penalty: float = 1.2) -> Any:
     """
     Predict a response using a client.
     """
-    client = Client(model_url)
-    response = client.predict(message, system_prompt, 0.7, 4096, 0.5, 1.2, api_name=("/chat_1" if model_url== models["Llama2-70b"]["chat_link"] else '/chat'))
+    client = Client("https://osanseviero-mistral-super-fast.hf.space/")
+    response = client.predict(
+        message,
+        temperature,
+        max_new_tokens,
+        topp,
+        repetition_penalty,
+        api_name="/chat"
+    )
     return response
 
 PATH = os.path.join(os.path.expanduser("~"), "Data")
@@ -114,11 +120,9 @@ def initialize_session_state():
 
 def sidebar():
     with st.sidebar:
-        st.sidebar.image('qr.jpg')
         st.markdown("# 💸 **Support our project**")
         st.markdown("This money would be used for paying for API and supporting our team.")
-        st.markdown("🎯 **Goal**: 600 $")
-        st.markdown("[🔗 Link to the bank](https://send.monobank.ua/jar/4mvqDivxmP)")
+        st.markdown("[🔗 Link](https://send.monobank.ua/jar/4mvqDivxmP)")
         
         st.markdown("")
 
@@ -135,7 +139,6 @@ def sidebar():
 
             # Embed the video
             st.markdown(embed_html, unsafe_allow_html=True)
-
 sidebar()
 
 initialize_session_state()
@@ -162,7 +165,7 @@ class LlamaLLM(LLM):
 
     def _call(self, prompt: str, stop: Optional[List[str]] = None, run_manager: Optional[CallbackManagerForLLMRun] = None) -> str:
         model_link = models[st.session_state.model_choice]["chat_link"]
-        response = predict(prompt, model_url=model_link)
+        response = predict(prompt)
         return response
 
     @property