alvanlii commited on
Commit
6405b6b
1 Parent(s): 3b033c6

Fix model path

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -19,10 +19,9 @@ MODEL_DIR = './fromage_model/fromage_vis4'
19
 
20
  class ChatBotCheese:
21
  def __init__(self):
22
- # from huggingface_hub import hf_hub_download
23
- # model_ckpt_path = hf_hub_download("alvanlii/fromage", "pretrained_ckpt.pth.tar")
24
- # self.model = models.load_fromage(MODEL_DIR, model_ckpt_path)
25
- self.model = models.load_fromage(MODEL_DIR)
26
  self.curr_image = None
27
 
28
  def add_image(self, state, image_in):
 
19
 
20
  class ChatBotCheese:
21
  def __init__(self):
22
+ from huggingface_hub import hf_hub_download
23
+ model_ckpt_path = hf_hub_download("alvanlii/fromage", "pretrained_ckpt.pth.tar")
24
+ self.model = models.load_fromage(MODEL_DIR, model_ckpt_path)
 
25
  self.curr_image = None
26
 
27
  def add_image(self, state, image_in):