Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -39,7 +39,7 @@ def predict(img):
|
|
39 |
|
40 |
# Create a prediction label and prediction probability dictionary for each prediction class
|
41 |
# This is the required format for Gradio's output parameter
|
42 |
-
pred_labels = '
|
43 |
|
44 |
# Calculate the prediction time
|
45 |
pred_time = round(timer() - start_time, 5)
|
|
|
39 |
|
40 |
# Create a prediction label and prediction probability dictionary for each prediction class
|
41 |
# This is the required format for Gradio's output parameter
|
42 |
+
pred_labels = 'Cat' if predicted.item() ==1 else 'Dog'
|
43 |
|
44 |
# Calculate the prediction time
|
45 |
pred_time = round(timer() - start_time, 5)
|