tsereno commited on
Commit
067af94
1 Parent(s): a150914

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -2
app.py CHANGED
@@ -95,10 +95,19 @@ def bot(
95
  else:
96
  #video_prompt = preprocess_image(video_prompt)
97
  model = genai.GenerativeModel('gemini-1.5-flash-latest')
 
 
 
 
 
 
 
 
98
  response = model.generate_content(
99
- contents=[text_prompt, video_prompt],
100
  stream=True,
101
- generation_config=generation_config)
 
102
  response.resolve()
103
 
104
 
 
95
  else:
96
  #video_prompt = preprocess_image(video_prompt)
97
  model = genai.GenerativeModel('gemini-1.5-flash-latest')
98
+
99
+ while video_file.state.name == "PROCESSING":
100
+ print('.', end='')
101
+ time.sleep(10)
102
+ video_file = genai.get_file(video_prompt)
103
+
104
+ if video_file.state.name == "FAILED":
105
+ raise ValueError(video_prompt)
106
  response = model.generate_content(
107
+ contents=[text_prompt, video_file],
108
  stream=True,
109
+ generation_config=generation_config,
110
+ request_options={"timeout": 600})
111
  response.resolve()
112
 
113