Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,7 @@
|
|
1 |
import os
|
2 |
import time
|
3 |
from typing import List, Tuple, Optional
|
|
|
4 |
|
5 |
import google.generativeai as genai
|
6 |
import gradio as gr
|
@@ -46,6 +47,7 @@ def bot(
|
|
46 |
model_name: str,
|
47 |
image_prompt: Optional[Image.Image],
|
48 |
video_prompt,
|
|
|
49 |
temperature: float,
|
50 |
max_output_tokens: int,
|
51 |
stop_sequences: str,
|
@@ -124,7 +126,8 @@ google_key_component = gr.Textbox(
|
|
124 |
)
|
125 |
|
126 |
image_prompt_component = gr.Image(type="pil", label="Image", scale=1)
|
127 |
-
video_prompt_component = gr.Video()
|
|
|
128 |
|
129 |
model_selection = gr.Dropdown(["gemini-1.0-pro", "gemini-pro-vision","gemini-1.5-flash-latest", "gemini-1.5-pro-latest","gemini-1.0-pro-001"],label="Select Gemini Model",value="gemini-1.0-pro")
|
130 |
|
@@ -205,6 +208,7 @@ bot_inputs = [
|
|
205 |
model_selection,
|
206 |
image_prompt_component,
|
207 |
video_prompt_component,
|
|
|
208 |
temperature_component,
|
209 |
max_output_tokens_component,
|
210 |
stop_sequences_component,
|
@@ -223,6 +227,7 @@ with gr.Blocks() as demo:
|
|
223 |
with gr.Row():
|
224 |
image_prompt_component.render()
|
225 |
video_prompt_component.render()
|
|
|
226 |
model_selection.render()
|
227 |
chatbot_component.render()
|
228 |
text_prompt_component.render()
|
|
|
1 |
import os
|
2 |
import time
|
3 |
from typing import List, Tuple, Optional
|
4 |
+
from pathlib import Path
|
5 |
|
6 |
import google.generativeai as genai
|
7 |
import gradio as gr
|
|
|
47 |
model_name: str,
|
48 |
image_prompt: Optional[Image.Image],
|
49 |
video_prompt,
|
50 |
+
file_prompt,
|
51 |
temperature: float,
|
52 |
max_output_tokens: int,
|
53 |
stop_sequences: str,
|
|
|
126 |
)
|
127 |
|
128 |
image_prompt_component = gr.Image(type="pil", label="Image", scale=1)
|
129 |
+
video_prompt_component = gr.Video(label="Video")
|
130 |
+
file_prompt_component = gr.UploadButton("Upload a file", file_count="directory", label="File")
|
131 |
|
132 |
model_selection = gr.Dropdown(["gemini-1.0-pro", "gemini-pro-vision","gemini-1.5-flash-latest", "gemini-1.5-pro-latest","gemini-1.0-pro-001"],label="Select Gemini Model",value="gemini-1.0-pro")
|
133 |
|
|
|
208 |
model_selection,
|
209 |
image_prompt_component,
|
210 |
video_prompt_component,
|
211 |
+
file_prompt_component,
|
212 |
temperature_component,
|
213 |
max_output_tokens_component,
|
214 |
stop_sequences_component,
|
|
|
227 |
with gr.Row():
|
228 |
image_prompt_component.render()
|
229 |
video_prompt_component.render()
|
230 |
+
file_prompt_component.render()
|
231 |
model_selection.render()
|
232 |
chatbot_component.render()
|
233 |
text_prompt_component.render()
|