arjunvankani commited on
Commit
785598b
·
verified ·
1 Parent(s): 64e13c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ qa = pipeline("question-answering", model="distilbert-base-cased-distilled-squad
12
  # --- Vision pipelines ---
13
  img_clf = pipeline("image-classification", model="google/vit-base-patch16-224")
14
  det = pipeline("object-detection", model="facebook/detr-resnet-50")
15
- seg = pipeline("image-segmentation", model="facebook/mask2former-swin-large-coco")
16
 
17
  # --- Diffusion model for text-to-image ---
18
  sd_pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
 
12
  # --- Vision pipelines ---
13
  img_clf = pipeline("image-classification", model="google/vit-base-patch16-224")
14
  det = pipeline("object-detection", model="facebook/detr-resnet-50")
15
+ seg = pipeline("image-segmentation", model="facebook/mask2former-swin-base-coco-instance")
16
 
17
  # --- Diffusion model for text-to-image ---
18
  sd_pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)