prithivMLmods commited on
Commit
bc87f11
·
verified ·
1 Parent(s): 0a8e8e5

update app

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -114,7 +114,7 @@ except Exception as e:
114
  model_x = None
115
  processor_x = None
116
 
117
- print("🔄 Loading Holo2-8B...")
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 == "Holo2-8B":
353
  model, processor = model_h, processor_h
354
  if model is None: return "Error: Holo2-8B model failed to load.", None
355
- print("Using Holo2-8B Pipeline...")
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", "Holo2-8B"],
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?", "Holo2-8B"],
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"