Update app.py

app.py

@@ -90,21 +90,36 @@ 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 = '', 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("https://osanseviero-mistral-super-fast.hf.space/")
-    response = client.predict(
-        message,
-        temperature,
-        max_new_tokens,
-        topp,
-        repetition_penalty,
-        api_name="/chat"
-    )
-    return response
+# 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("https://osanseviero-mistral-super-fast.hf.space/")
+#     response = client.predict(
+#         message,
+#         temperature,
+#         max_new_tokens,
+#         topp,
+#         repetition_penalty,
+#         api_name="/chat"
+#     )
+#     return response
+
+def predict(input, path = None):
+  client = Client("https://roboflow-gemini.hf.space/--replicas/bkd57/")
+  result = client.predict(
+      None,	
+      [path],	
+      0.4,	
+      2048,	
+      "",	
+      32,	
+      1,	
+      [[input,None]],
+                api_name="/bot"
+  )
+  return result[0][1]
 
 PATH = os.path.join(os.path.expanduser("~"), "Data")