LukasHug commited on
Commit
ce1a545
·
verified ·
1 Parent(s): dda3db7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -584,7 +584,6 @@ model_path = DEFAULT_MODEL
584
  logger.info(f"Loading model: {model_path}")
585
  # Check if it's a Qwen model
586
  if "qwenguard" in model_path.lower():
587
- @spaces.GPU
588
  model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
589
  model_path,
590
  torch_dtype="auto",
@@ -595,7 +594,6 @@ if "qwenguard" in model_path.lower():
595
 
596
  # Otherwise assume it's a LlavaGuard model
597
  else:
598
- @spaces.GPU
599
  model = LlavaOnevisionForConditionalGeneration.from_pretrained(
600
  model_path,
601
  torch_dtype="auto",
 
584
  logger.info(f"Loading model: {model_path}")
585
  # Check if it's a Qwen model
586
  if "qwenguard" in model_path.lower():
 
587
  model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
588
  model_path,
589
  torch_dtype="auto",
 
594
 
595
  # Otherwise assume it's a LlavaGuard model
596
  else:
 
597
  model = LlavaOnevisionForConditionalGeneration.from_pretrained(
598
  model_path,
599
  torch_dtype="auto",