Update app.py

app.py

@@ -7,32 +7,32 @@ from transformers import AutoFeatureExtractor, AutoModelForImageClassification
 
 extractor = AutoFeatureExtractor.from_pretrained("rwcuffney/autotrain-pick_a_card-3726099221")
 model = AutoModelForImageClassification.from_pretrained("rwcuffney/autotrain-pick_a_card-3726099221")
-st.write(model.name)
+st.write(model.__class__.__name__)
 st.code(type(model))
 
 
 extractor = AutoFeatureExtractor.from_pretrained("rwcuffney/autotrain-pick_a_card-3726099222")
 model = AutoModelForImageClassification.from_pretrained("rwcuffney/autotrain-pick_a_card-3726099222")
-st.write(model.name)
+st.write(model.__class__.__name__)
 st.code(type(model))
 
 
 extractor = AutoFeatureExtractor.from_pretrained("rwcuffney/autotrain-pick_a_card-3726099223")
 model = AutoModelForImageClassification.from_pretrained("rwcuffney/autotrain-pick_a_card-3726099223")
-st.write(model.name)
+st.write(model.__class__.__name__)
 st.code(type(model))
 
 
 
 extractor = AutoFeatureExtractor.from_pretrained("rwcuffney/autotrain-pick_a_card-3726099224")
 model = AutoModelForImageClassification.from_pretrained("rwcuffney/autotrain-pick_a_card-3726099224")
-st.write(model.name)
+st.write(model.__class__.__name__)
 st.code(type(model))
 
 
 extractor = AutoFeatureExtractor.from_pretrained("rwcuffney/autotrain-pick_a_card-3726099225")
 model = AutoModelForImageClassification.from_pretrained("rwcuffney/autotrain-pick_a_card-3726099225")
-st.write(model.name)
+st.write(model.__class__.__name__)
 st.code(type(model))