wd-tagger

Running

SmilingWolf commited on Mar 6

Commit

038b1de

•

1 Parent(s): 2375aa3

Update app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -129,7 +129,7 @@ class Predictor:
         _, height, width, _ = model.get_inputs()[0].shape
         self.model_target_size = height
-        self.last_loaded_repo = model_path
         self.model = model
     def prepare_image(self, image):

         _, height, width, _ = model.get_inputs()[0].shape
         self.model_target_size = height
+        self.last_loaded_repo = model_repo
         self.model = model
     def prepare_image(self, image):