Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -205,8 +205,9 @@ with gr.Blocks(title="Ovis-U1-3B") as demo:
|
|
| 205 |
output_gallery = gr.Gallery(label="Generated Images", columns=2, visible=True) # Default to visible, content will control
|
| 206 |
output_text = gr.Textbox(label="Generated Text", visible=False, lines=5, interactive=False)
|
| 207 |
|
| 208 |
-
@spaces.GPU
|
| 209 |
def run_img_txt_to_img_tab(prompt, img, steps, seed, txt_cfg, img_cfg, progress=gr.Progress(track_tqdm=True)):
|
|
|
|
| 210 |
if img is None:
|
| 211 |
return (
|
| 212 |
gr.update(value=[], visible=False),
|
|
@@ -219,7 +220,8 @@ with gr.Blocks(title="Ovis-U1-3B") as demo:
|
|
| 219 |
gr.update(value="", visible=False)
|
| 220 |
)
|
| 221 |
|
| 222 |
-
@spaces.GPU
|
|
|
|
| 223 |
def run_txt_to_img_tab(prompt, height, width, steps, seed, guidance, progress=gr.Progress(track_tqdm=True)):
|
| 224 |
# Seed is already finalized by the randomize_seed_fn in the click chain
|
| 225 |
imgs = process_txt_to_img(prompt, height, width, steps, seed, guidance, progress=progress)
|
|
@@ -228,7 +230,8 @@ with gr.Blocks(title="Ovis-U1-3B") as demo:
|
|
| 228 |
gr.update(value="", visible=False)
|
| 229 |
)
|
| 230 |
|
| 231 |
-
@spaces.GPU
|
|
|
|
| 232 |
def run_img_to_txt_tab(img, prompt, progress=gr.Progress(track_tqdm=True)):
|
| 233 |
if img is None:
|
| 234 |
return (
|
|
@@ -341,4 +344,4 @@ with gr.Blocks(title="Ovis-U1-3B") as demo:
|
|
| 341 |
)
|
| 342 |
|
| 343 |
if __name__ == "__main__":
|
| 344 |
-
demo.launch(
|
|
|
|
| 205 |
output_gallery = gr.Gallery(label="Generated Images", columns=2, visible=True) # Default to visible, content will control
|
| 206 |
output_text = gr.Textbox(label="Generated Text", visible=False, lines=5, interactive=False)
|
| 207 |
|
| 208 |
+
@spaces.GPU(duration=20)
|
| 209 |
def run_img_txt_to_img_tab(prompt, img, steps, seed, txt_cfg, img_cfg, progress=gr.Progress(track_tqdm=True)):
|
| 210 |
+
"""Use Ovis-U1-3B to modify an image. Supply Image and Text Prompt"""
|
| 211 |
if img is None:
|
| 212 |
return (
|
| 213 |
gr.update(value=[], visible=False),
|
|
|
|
| 220 |
gr.update(value="", visible=False)
|
| 221 |
)
|
| 222 |
|
| 223 |
+
@spaces.GPU(duration=20)
|
| 224 |
+
"""Use Ovis-U1-3B to generate an Image."""
|
| 225 |
def run_txt_to_img_tab(prompt, height, width, steps, seed, guidance, progress=gr.Progress(track_tqdm=True)):
|
| 226 |
# Seed is already finalized by the randomize_seed_fn in the click chain
|
| 227 |
imgs = process_txt_to_img(prompt, height, width, steps, seed, guidance, progress=progress)
|
|
|
|
| 230 |
gr.update(value="", visible=False)
|
| 231 |
)
|
| 232 |
|
| 233 |
+
@spaces.GPU(duration=20)
|
| 234 |
+
"""User Ovis-U1-3B to understand an Image (Vision processing)"""
|
| 235 |
def run_img_to_txt_tab(img, prompt, progress=gr.Progress(track_tqdm=True)):
|
| 236 |
if img is None:
|
| 237 |
return (
|
|
|
|
| 344 |
)
|
| 345 |
|
| 346 |
if __name__ == "__main__":
|
| 347 |
+
demo.launch(mcp_server=True)
|