Update app.py
Browse files
app.py
CHANGED
@@ -47,7 +47,7 @@ from moviepy.editor import VideoFileClip, concatenate_videoclips
|
|
47 |
|
48 |
def bot(
|
49 |
google_key: str,
|
50 |
-
|
51 |
image_prompt: Optional[Image.Image],
|
52 |
video_prompt: List[Path],
|
53 |
temperature: float,
|
@@ -83,7 +83,7 @@ def bot(
|
|
83 |
else:
|
84 |
video_path = str(video_prompt[0])
|
85 |
|
86 |
-
model = genai.GenerativeModel(
|
87 |
|
88 |
video_file = genai.upload_file(path=video_path)
|
89 |
while video_file.state.name == "PROCESSING":
|
@@ -102,14 +102,14 @@ def bot(
|
|
102 |
response.resolve()
|
103 |
elif image_prompt is not None:
|
104 |
image_prompt = preprocess_image(image_prompt)
|
105 |
-
model = genai.GenerativeModel(
|
106 |
response = model.generate_content(
|
107 |
contents=[text_prompt, image_prompt],
|
108 |
stream=True,
|
109 |
generation_config=generation_config)
|
110 |
response.resolve()
|
111 |
else:
|
112 |
-
model = genai.GenerativeModel(
|
113 |
response = model.generate_content(
|
114 |
text_prompt,
|
115 |
stream=True,
|
@@ -213,7 +213,6 @@ user_inputs = [text_prompt_component,
|
|
213 |
|
214 |
bot_inputs = [
|
215 |
google_key_component,
|
216 |
-
model_selection,
|
217 |
image_prompt_component,
|
218 |
video_prompt_component,
|
219 |
temperature_component,
|
@@ -232,7 +231,6 @@ with gr.Blocks() as demo:
|
|
232 |
with gr.Row():
|
233 |
image_prompt_component.render()
|
234 |
video_prompt_component.render()
|
235 |
-
model_selection.render()
|
236 |
chatbot_component.render()
|
237 |
text_prompt_component.render()
|
238 |
run_button_component.render()
|
|
|
47 |
|
48 |
def bot(
|
49 |
google_key: str,
|
50 |
+
|
51 |
image_prompt: Optional[Image.Image],
|
52 |
video_prompt: List[Path],
|
53 |
temperature: float,
|
|
|
83 |
else:
|
84 |
video_path = str(video_prompt[0])
|
85 |
|
86 |
+
model = genai.GenerativeModel("gemini-1.5-pro-latest")
|
87 |
|
88 |
video_file = genai.upload_file(path=video_path)
|
89 |
while video_file.state.name == "PROCESSING":
|
|
|
102 |
response.resolve()
|
103 |
elif image_prompt is not None:
|
104 |
image_prompt = preprocess_image(image_prompt)
|
105 |
+
model = genai.GenerativeModel("gemini-1.5-pro-latest")
|
106 |
response = model.generate_content(
|
107 |
contents=[text_prompt, image_prompt],
|
108 |
stream=True,
|
109 |
generation_config=generation_config)
|
110 |
response.resolve()
|
111 |
else:
|
112 |
+
model = genai.GenerativeModel("gemini-1.5-pro-latest")
|
113 |
response = model.generate_content(
|
114 |
text_prompt,
|
115 |
stream=True,
|
|
|
213 |
|
214 |
bot_inputs = [
|
215 |
google_key_component,
|
|
|
216 |
image_prompt_component,
|
217 |
video_prompt_component,
|
218 |
temperature_component,
|
|
|
231 |
with gr.Row():
|
232 |
image_prompt_component.render()
|
233 |
video_prompt_component.render()
|
|
|
234 |
chatbot_component.render()
|
235 |
text_prompt_component.render()
|
236 |
run_button_component.render()
|