Spaces:
Running
on
Zero
Running
on
Zero
update app
Browse files
app.py
CHANGED
|
@@ -114,7 +114,7 @@ except Exception as e:
|
|
| 114 |
model_x = None
|
| 115 |
processor_x = None
|
| 116 |
|
| 117 |
-
print("🔄 Loading
|
| 118 |
MODEL_ID_H = "Hcompany/Holo1-3B"
|
| 119 |
try:
|
| 120 |
processor_h = AutoProcessor.from_pretrained(MODEL_ID_H, trust_remote_code=True)
|
|
@@ -349,10 +349,10 @@ def process_screenshot(input_numpy_image: np.ndarray, task: str, model_choice: s
|
|
| 349 |
model, processor = model_x, processor_x
|
| 350 |
if model is None: return "Error: UI-TARS model failed to load.", None
|
| 351 |
print("Using UI-TARS Pipeline...")
|
| 352 |
-
elif model_choice == "
|
| 353 |
model, processor = model_h, processor_h
|
| 354 |
if model is None: return "Error: Holo2-8B model failed to load.", None
|
| 355 |
-
print("Using
|
| 356 |
else:
|
| 357 |
return f"Error: Unknown model '{model_choice}'", None
|
| 358 |
|
|
@@ -414,7 +414,7 @@ with gr.Blocks() as demo:
|
|
| 414 |
|
| 415 |
with gr.Row():
|
| 416 |
model_choice = gr.Radio(
|
| 417 |
-
choices=["Fara-7B", "UI-TARS-1.5-7B", "
|
| 418 |
label="Select Model",
|
| 419 |
value="Fara-7B",
|
| 420 |
interactive=True
|
|
@@ -441,7 +441,7 @@ with gr.Blocks() as demo:
|
|
| 441 |
examples=[
|
| 442 |
["./assets/google.png", "Search for 'Hugging Face'", "Fara-7B"],
|
| 443 |
["./assets/google.png", "Click the microphone button", "UI-TARS-1.5-7B"],
|
| 444 |
-
["./assets/google.png", "Where is the 'I'm Feeling Lucky' button?", "
|
| 445 |
],
|
| 446 |
inputs=[input_image, task_input, model_choice],
|
| 447 |
label="Quick Examples"
|
|
|
|
| 114 |
model_x = None
|
| 115 |
processor_x = None
|
| 116 |
|
| 117 |
+
print("🔄 Loading Holo1-3B...")
|
| 118 |
MODEL_ID_H = "Hcompany/Holo1-3B"
|
| 119 |
try:
|
| 120 |
processor_h = AutoProcessor.from_pretrained(MODEL_ID_H, trust_remote_code=True)
|
|
|
|
| 349 |
model, processor = model_x, processor_x
|
| 350 |
if model is None: return "Error: UI-TARS model failed to load.", None
|
| 351 |
print("Using UI-TARS Pipeline...")
|
| 352 |
+
elif model_choice == "Holo1-3B":
|
| 353 |
model, processor = model_h, processor_h
|
| 354 |
if model is None: return "Error: Holo2-8B model failed to load.", None
|
| 355 |
+
print("Using Holo1-3B Pipeline...")
|
| 356 |
else:
|
| 357 |
return f"Error: Unknown model '{model_choice}'", None
|
| 358 |
|
|
|
|
| 414 |
|
| 415 |
with gr.Row():
|
| 416 |
model_choice = gr.Radio(
|
| 417 |
+
choices=["Fara-7B", "UI-TARS-1.5-7B", "Holo1-3B"],
|
| 418 |
label="Select Model",
|
| 419 |
value="Fara-7B",
|
| 420 |
interactive=True
|
|
|
|
| 441 |
examples=[
|
| 442 |
["./assets/google.png", "Search for 'Hugging Face'", "Fara-7B"],
|
| 443 |
["./assets/google.png", "Click the microphone button", "UI-TARS-1.5-7B"],
|
| 444 |
+
["./assets/google.png", "Where is the 'I'm Feeling Lucky' button?", "Holo1-3B"],
|
| 445 |
],
|
| 446 |
inputs=[input_image, task_input, model_choice],
|
| 447 |
label="Quick Examples"
|