Hugging Face's logo

Spaces:
AlexHung29629
/
UI_Screen_Description_Generator_with_Pix2Struct
Sleeping

App Files Community
AlexHung29629 commited on
Commit
89612fd
·
verified ·
1 Parent(s): b9f48bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from transformers import Pix2StructForConditionalGeneration, Pix2StructProcessor
6
 
7
  # Load model and processor
8
  model = Pix2StructForConditionalGeneration.from_pretrained("google/pix2struct-screen2words-large", torch_dtype=torch.bfloat16).to("cuda")
9
- processor = Pix2StructProcessor.from_pretrained("google/pix2struct-screen2words-large)
10
 
11
  # Define the function
12
  @spaces.GPU
 
6
 
7
  # Load model and processor
8
  model = Pix2StructForConditionalGeneration.from_pretrained("google/pix2struct-screen2words-large", torch_dtype=torch.bfloat16).to("cuda")
9
+ processor = Pix2StructProcessor.from_pretrained("google/pix2struct-screen2words-large")
10
 
11
  # Define the function
12
  @spaces.GPU