davidr99 commited on
Commit
a8bbcc2
·
verified ·
1 Parent(s): 6042dcf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -19,7 +19,9 @@ finetuned_model_id = "davidr99/finetuned_paligemma_blackjack"
19
 
20
  import gradio as gr
21
 
22
- model = PaliGemmaForConditionalGeneration.from_pretrained(finetuned_model_id)
 
 
23
  processor = AutoProcessor.from_pretrained(model_id)
24
 
25
  def blackjack_ai(image):
 
19
 
20
  import gradio as gr
21
 
22
+ model = PaliGemmaForConditionalGeneration.from_pretrained(model_id)
23
+ model.load_adapter(finetuned_model_id)
24
+
25
  processor = AutoProcessor.from_pretrained(model_id)
26
 
27
  def blackjack_ai(image):