Spaces:

pathananas
/

multimodal-ai-engine

Sleeping

pathananas commited on 24 days ago

Commit

56a2edf

verified ·

1 Parent(s): 5a2b743

model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -34,7 +34,7 @@ topic_pipeline = pipeline(
 image_pipeline = pipeline(
     "image-classification",
-    model="facebook/convnext-base-224",
     device=device,
     batch_size=8
 )

 image_pipeline = pipeline(
     "image-classification",
+    model="google/vit-base-patch16-224",
     device=device,
     batch_size=8
 )