Update app.py
Browse files
app.py
CHANGED
|
@@ -62,7 +62,7 @@ def get_attn_map(image, attn_scores, n_width, n_height):
|
|
| 62 |
# load model
|
| 63 |
if torch.cuda.is_available():
|
| 64 |
# os.system('pip install flash-attn --no-build-isolation')
|
| 65 |
-
model_name_or_path = "microsoft/GUI-Actor-
|
| 66 |
data_processor = AutoProcessor.from_pretrained(model_name_or_path)
|
| 67 |
tokenizer = data_processor.tokenizer
|
| 68 |
model = Qwen2_5_VLForConditionalGenerationWithPointer.from_pretrained(
|
|
|
|
| 62 |
# load model
|
| 63 |
if torch.cuda.is_available():
|
| 64 |
# os.system('pip install flash-attn --no-build-isolation')
|
| 65 |
+
model_name_or_path = "microsoft/GUI-Actor-3B-Qwen2.5-VL"
|
| 66 |
data_processor = AutoProcessor.from_pretrained(model_name_or_path)
|
| 67 |
tokenizer = data_processor.tokenizer
|
| 68 |
model = Qwen2_5_VLForConditionalGenerationWithPointer.from_pretrained(
|