hi

app.py

@@ -33,15 +33,15 @@ char = {"Men": "020101-020139", "Women": "020301-020328", "Children": "020501-02
 # Function to predict outputs
 def predict(img, model_name):
     if (model_name == 'Geometric Figures & Solids'):
-        geo_learn = from_pretrained_fastai('amandasarubbi/geo_model', cpu=True)
+        geo_learn = from_pretrained_fastai('amandasarubbi/geo_model')
         pred = geo_learn.predict(img)[0]
         labels = geo
     elif (model_name == 'Scenery, Natural Phenomena'):
-        landscape_learn = from_pretrained_fastai('amandasarubbi/landscape_model', cpu=True)
+        landscape_learn = from_pretrained_fastai('amandasarubbi/landscape_model')
         pred = landscape_learn.predict(img)[0]
         labels = scen
     elif (model_name == 'Human & Supernatural Beings'):
-        human_learn = from_pretrained_fastai('amandasarubbi/human_model', cpu=True)
+        human_learn = from_pretrained_fastai('amandasarubbi/human_model')
         pred = human_learn.predict(img)[0]
         labels = char