prithivMLmods commited on
Commit
455a710
·
verified ·
1 Parent(s): 0358302

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,7 +11,7 @@ def select_emotion():
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,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 == "dog_breed":
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"}