Update app.py

app.py

@@ -42,8 +42,8 @@ class mm_inference:
         self.img_feature_extractor = AutoFeatureExtractor.from_pretrained(self.img_model_ckpt)
 
         # text and image pipeles for feature extraction
-        self.text_feature_extractor = pipeline(task = 'feature-extraction', model = 'lzun/spanish-social-media-boxing-text', tokenizer = self.tokenizer, return_tensors = True, device = device)
-        self.img_model = ViTModel.from_pretrained('lzun/spanish-social-media-boxing-images')
+        self.text_feature_extractor = pipeline(task = 'feature-extraction', model = 'lzun/mcovmex-text', tokenizer = self.tokenizer, return_tensors = True, device = device)
+        self.img_model = ViTModel.from_pretrained('lzun/mcovmex-image')
 
     def get_text_embs(self, text):
         '''