davidr99 commited on
Commit
ae410c3
·
verified ·
1 Parent(s): 03561cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -26,8 +26,8 @@ def blackjack_ai(image):
26
  global model, processor
27
 
28
  if model == None:
29
- model = PaliGemmaForConditionalGeneration.from_pretrained(finetuned_model_id)
30
- processor = AutoProcessor.from_pretrained(model_id)
31
 
32
  prompt = "<image><bos>extract json\n"
33
  image_file = Image.fromarray(image.astype('uint8'), 'RGB')
 
26
  global model, processor
27
 
28
  if model == None:
29
+ model = PaliGemmaForConditionalGeneration.from_pretrained(model_id)
30
+ processor = AutoProcessor.from_pretrained(finetuned_model_id)
31
 
32
  prompt = "<image><bos>extract json\n"
33
  image_file = Image.fromarray(image.astype('uint8'), 'RGB')