Illia56 commited on
Commit
03e52ca
1 Parent(s): d92b221

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -106,11 +106,11 @@ def transcribe_video(youtube_url: str, path: str) -> List[Document]:
106
  # )
107
  # return response
108
 
109
- def predict(input, path = None):
110
  client = Client("https://roboflow-gemini.hf.space/--replicas/bkd57/")
111
  result = client.predict(
112
  None,
113
- [path],
114
  0.4,
115
  2048,
116
  "",
@@ -120,6 +120,7 @@ def predict(input, path = None):
120
  api_name="/bot"
121
  )
122
  return result[0][1]
 
123
 
124
  PATH = os.path.join(os.path.expanduser("~"), "Data")
125
 
 
106
  # )
107
  # return response
108
 
109
+ def predict(input, images = []):
110
  client = Client("https://roboflow-gemini.hf.space/--replicas/bkd57/")
111
  result = client.predict(
112
  None,
113
+ images,
114
  0.4,
115
  2048,
116
  "",
 
120
  api_name="/bot"
121
  )
122
  return result[0][1]
123
+
124
 
125
  PATH = os.path.join(os.path.expanduser("~"), "Data")
126