Fix model path
Browse files
app.py
CHANGED
@@ -19,10 +19,9 @@ MODEL_DIR = './fromage_model/fromage_vis4'
|
|
19 |
|
20 |
class ChatBotCheese:
|
21 |
def __init__(self):
|
22 |
-
|
23 |
-
|
24 |
-
|
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):
|