Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -34,5 +34,7 @@ def predict(image):
|
|
34 |
label = model.config.id2label[predicted_class_idx].split(",")[0]
|
35 |
time.sleep(2)
|
36 |
return {label: float(predicted_class_prob)}
|
37 |
-
|
38 |
-
|
|
|
|
|
|
34 |
label = model.config.id2label[predicted_class_idx].split(",")[0]
|
35 |
time.sleep(2)
|
36 |
return {label: float(predicted_class_prob)}
|
37 |
+
|
38 |
+
|
39 |
+
examples = [['data/closed_highrise.jpg'], ['data/open_lowrise.jpg'],['data/dense_trees.jpg'],['data/large_lowrise.jpg']]
|
40 |
+
gr.Interface(predict, gr.Image(type="pil"), "label", examples=examples).launch()
|