ShreyMehra commited on
Commit
8aad5c5
·
unverified ·
1 Parent(s): a50069f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -47,7 +47,7 @@ class UI:
47
 
48
 
49
  class Model:
50
- def load_model():
51
  peft_model_id = "Shrey23/Image-Captioning"
52
  config = PeftConfig.from_pretrained(peft_model_id)
53
  global model
 
47
 
48
 
49
  class Model:
50
+ def load_model(self):
51
  peft_model_id = "Shrey23/Image-Captioning"
52
  config = PeftConfig.from_pretrained(peft_model_id)
53
  global model