Spaces:

tomeheya
/

IM-417-classification

Sleeping

storesource commited on Apr 10

Commit

f1f8687

•

1 Parent(s): 2c9bd78

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,8 +15,8 @@ from transformers import AutoModel
 #@st.cache(allow_output_mutation=True)  # Allow model mutation for prediction
 def load_model():
     try:
-        # model = from_pretrained_keras("tomeheya/IM-417-128x128-classification", force_download=True)
-        model = AutoModel.from_pretrained("tomeheya/IM-417-128x128-classification")
         print("Model loaded successfully!")
         return model
     except Exception as e:

 #@st.cache(allow_output_mutation=True)  # Allow model mutation for prediction
 def load_model():
     try:
+        model = from_pretrained_keras("tomeheya/IM-417-128x128-classification", force_download=True)
+        #model = AutoModel.from_pretrained("tomeheya/IM-417-128x128-classification")
         print("Model loaded successfully!")
         return model
     except Exception as e: