Update app.py
Browse files
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,
|
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 |
|