Spaces:

chlab
/

interactive_kinematic_planet_detector

Sleeping

jpterry commited on Jun 16, 2022

Commit

8514431

1 Parent(s): 096b8bd

just automodel

Files changed (1) hide show

app.py CHANGED Viewed

@@ -270,7 +270,7 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
     # model = EfficientNetPreTrained.from_pretrained(model_loading_name)
     # model = AutoModel.from_pretrained(model_loading_name, trust_remote_code=True)
-    model = AutoModelForImageClassification.from_pretrained(model_path + model_loading_name)
     # model = EfficientNet(dropout=hparams.dropout,
     #                                 num_channels=hparams.num_channels,

     # model = EfficientNetPreTrained.from_pretrained(model_loading_name)
     # model = AutoModel.from_pretrained(model_loading_name, trust_remote_code=True)
+    model = AutoModel.from_pretrained(model_path + model_loading_name)
     # model = EfficientNet(dropout=hparams.dropout,
     #                                 num_channels=hparams.num_channels,