Update app.py

app.py

@@ -3,7 +3,12 @@ import pandas as pd
 import matplotlib.pyplot as plt
 
 
+from transformers import AutoFeatureExtractor, AutoModelForImageClassification
+extractor = AutoFeatureExtractor.from_pretrained("rwcuffney/autotrain-pick_a_card-3726099223")
+model = AutoModelForImageClassification.from_pretrained("rwcuffney/autotrain-pick_a_card-3726099223")
+st.code(model)
 
+'''
 x = st.slider('Select a value')
 st.write(x, 'squared is', x * x)
 
@@ -39,4 +44,5 @@ df_validate = df[df['data set']=='validate']
 from datasets import load_dataset #this isn't working
 dataset = load_dataset("rwcuffney/pick_a_card_test") #rwcuffney/pick_a_card_test
 
-st.write(df.head(20))
+st.write(df.head(20))
+'''