Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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-
|
| 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)
|