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(
|