Switch from MobileNet to CLIP

woctezuma · Jan 21, 2021 · 7045723 · 7045723
1 parent d8df4c2
commit 7045723
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model_utils.py b/model_utils.py
@@ -16,7 +16,7 @@
 )
 
 
-def get_my_model_of_choice(choice_index=0):
+def get_my_model_of_choice(choice_index=-1):
     available_models = [get_model_slug_for_keras(), get_model_slug_for_clip()]
 
     # The following line is where you can switch between Keras' MobileNet and OpenAI's CLIP: