Noeies commited on
Commit
4fc55ea
·
verified ·
1 Parent(s): 1ee8204

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -6,7 +6,11 @@ import matplotlib.pyplot as plt
6
  import numpy as np
7
 
8
  # โหลดโมเดล MedSAM2 จาก Hugging Face
9
- model_name = "MLforHealthcare/sam2rad"
 
 
 
 
10
  processor = AutoImageProcessor.from_pretrained(model_name)
11
  model = AutoModelForImageSegmentation.from_pretrained(model_name)
12
 
 
6
  import numpy as np
7
 
8
  # โหลดโมเดล MedSAM2 จาก Hugging Face
9
+ model_name = "from transformers import DetrImageProcessor, DetrForSegmentation
10
+
11
+ model_name = "facebook/detr-resnet-50-panoptic"
12
+ processor = DetrImageProcessor.from_pretrained(model_name)
13
+ model = DetrForSegmentation.from_pretrained(model_name)"
14
  processor = AutoImageProcessor.from_pretrained(model_name)
15
  model = AutoModelForImageSegmentation.from_pretrained(model_name)
16