emkessle commited on
Commit
bcad485
·
verified ·
1 Parent(s): b1ef62a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,7 +12,7 @@ import huggingface_hub # For downloading model assets
12
  import autogluon.multimodal # For loading AutoGluon image classifier
13
 
14
  # Hardcoded Hub model (native zip)
15
- MODEL_REPO_ID = "apsora/autoML_images_data"
16
  ZIP_FILENAME = "autogluon_image_predictor_dir.zip"
17
 
18
  # Local cache/extract dirs
@@ -93,7 +93,7 @@ with gradio.Blocks() as demo:
93
  # Provide an introduction
94
  gradio.Markdown("# Tomato or Not?")
95
  gradio.Markdown("""
96
- This is a simple app that uses the model at apsora/autoML_images_data to classify whether an image has a tomato or not,
97
  utilizing data found at Iris314/Food_tomatoes_dataset. To use the interface, upload an image in the area shown below.
98
  """)
99
 
 
12
  import autogluon.multimodal # For loading AutoGluon image classifier
13
 
14
  # Hardcoded Hub model (native zip)
15
+ MODEL_REPO_ID = "kaitongg/best_tomato_model"
16
  ZIP_FILENAME = "autogluon_image_predictor_dir.zip"
17
 
18
  # Local cache/extract dirs
 
93
  # Provide an introduction
94
  gradio.Markdown("# Tomato or Not?")
95
  gradio.Markdown("""
96
+ This is a simple app that uses the model at kaitongg/best_tomato_model to classify whether an image has a tomato or not,
97
  utilizing data found at Iris314/Food_tomatoes_dataset. To use the interface, upload an image in the area shown below.
98
  """)
99