Application file

app.py

@@ -32,15 +32,15 @@ if torch.cuda.is_available():
 else:
     device = torch.device("cpu")
 
-# model = Owlv2ForObjectDetection.from_pretrained("google/owlv2-base-patch16-ensemble").to(device)
-# processor = Owlv2Processor.from_pretrained("google/owlv2-base-patch16-ensemble")
-# model_sam = SamModel.from_pretrained("facebook/sam-vit-huge").to(device)
-# processor_sam = SamProcessor.from_pretrained("facebook/sam-vit-huge")
+model = Owlv2ForObjectDetection.from_pretrained("google/owlv2-base-patch16-ensemble").to(device)
+processor = Owlv2Processor.from_pretrained("google/owlv2-base-patch16-ensemble")
+model_sam = SamModel.from_pretrained("facebook/sam-vit-huge").to(device)
+processor_sam = SamProcessor.from_pretrained("facebook/sam-vit-huge")
     
-model = Owlv2ForObjectDetection.from_pretrained("owlv2-base-patch16-ensemble").to(device)
-processor = Owlv2Processor.from_pretrained("owlv2-base-patch16-ensemble")
-model_sam = SamModel.from_pretrained("SAM/sam-vit-huge").to(device)
-processor_sam = SamProcessor.from_pretrained("SAM/sam-vit-huge")
+# model = Owlv2ForObjectDetection.from_pretrained("owlv2-base-patch16-ensemble").to(device)
+# processor = Owlv2Processor.from_pretrained("owlv2-base-patch16-ensemble")
+# model_sam = SamModel.from_pretrained("SAM/sam-vit-huge").to(device)
+# processor_sam = SamProcessor.from_pretrained("SAM/sam-vit-huge")
 
 
 @spaces.GPU