Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ def select_emotion():
|
|
11 |
return "emotion"
|
12 |
|
13 |
def select_dog_breed():
|
14 |
-
return "
|
15 |
|
16 |
# Main classification function that calls the appropriate model based on selection.
|
17 |
def classify(image, model_name):
|
@@ -19,7 +19,7 @@ def classify(image, model_name):
|
|
19 |
return gender_classification(image)
|
20 |
elif model_name == "emotion":
|
21 |
return emotion_classification(image)
|
22 |
-
elif model_name == "
|
23 |
return dog_breed_classification(image)
|
24 |
else:
|
25 |
return {"Error": "No model selected"}
|
|
|
11 |
return "emotion"
|
12 |
|
13 |
def select_dog_breed():
|
14 |
+
return "dog breed"
|
15 |
|
16 |
# Main classification function that calls the appropriate model based on selection.
|
17 |
def classify(image, model_name):
|
|
|
19 |
return gender_classification(image)
|
20 |
elif model_name == "emotion":
|
21 |
return emotion_classification(image)
|
22 |
+
elif model_name == "dog breed":
|
23 |
return dog_breed_classification(image)
|
24 |
else:
|
25 |
return {"Error": "No model selected"}
|