Update model_2.py

model_2.py

@@ -11,8 +11,8 @@ import time
 DEVICE = "cpu"
 DEVICE = "cpu"
 # DEVICE = "cuda:0"
 # DEVICE = "cuda:0"
 # Load the Pix2Struct model and processor
 # Load the Pix2Struct model and processor
-model = Pix2StructForConditionalGeneration.from_pretrained("google/pix2struct-docvqa-base").to(DEVICE)
-processor = Pix2StructProcessor.from_pretrained("google/pix2struct-docvqa-base")
+model = Pix2StructForConditionalGeneration.from_pretrained("google/pix2struct-docvqa-large").to(DEVICE)
+processor = Pix2StructProcessor.from_pretrained("google/pix2struct-docvqa-large")
 
 
 def generate(img, questions):
 def generate(img, questions):
     inputs = processor(images=[img for _ in range(len(questions))],
     inputs = processor(images=[img for _ in range(len(questions))],