Spaces:
Sleeping
Sleeping
fix deprecation
Browse files
app.py
CHANGED
@@ -16,6 +16,7 @@ transform = create_transform(
|
|
16 |
)
|
17 |
model.eval()
|
18 |
|
|
|
19 |
def predict_fn(img):
|
20 |
img = img.convert('RGB')
|
21 |
img = transform(img).unsqueeze(0)
|
@@ -29,4 +30,5 @@ def predict_fn(img):
|
|
29 |
|
30 |
return {LABELS[i]: v.item() for i, v in zip(indices, values)}
|
31 |
|
32 |
-
|
|
|
|
16 |
)
|
17 |
model.eval()
|
18 |
|
19 |
+
|
20 |
def predict_fn(img):
|
21 |
img = img.convert('RGB')
|
22 |
img = transform(img).unsqueeze(0)
|
|
|
30 |
|
31 |
return {LABELS[i]: v.item() for i, v in zip(indices, values)}
|
32 |
|
33 |
+
|
34 |
+
gr.Interface(predict_fn, gr.components.Image(type='pil'), outputs='label').launch()
|