Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -37,9 +37,14 @@ st.code(type(model))
|
|
37 |
from transformers import AutoImageProcessor
|
38 |
import torch
|
39 |
|
|
|
|
|
|
|
|
|
40 |
image_processor = AutoImageProcessor.from_pretrained("rwcuffney/autotrain-pick_a_card-3726099224"))
|
41 |
inputs = image_processor(image, return_tensors="pt")
|
42 |
|
|
|
43 |
'''
|
44 |
extractor = AutoFeatureExtractor.from_pretrained("rwcuffney/autotrain-pick_a_card-3726099225")
|
45 |
model = AutoModelForImageClassification.from_pretrained("rwcuffney/autotrain-pick_a_card-3726099225")
|
|
|
37 |
from transformers import AutoImageProcessor
|
38 |
import torch
|
39 |
|
40 |
+
|
41 |
+
image = dataset["test"][0]
|
42 |
+
st.image(image)
|
43 |
+
|
44 |
image_processor = AutoImageProcessor.from_pretrained("rwcuffney/autotrain-pick_a_card-3726099224"))
|
45 |
inputs = image_processor(image, return_tensors="pt")
|
46 |
|
47 |
+
|
48 |
'''
|
49 |
extractor = AutoFeatureExtractor.from_pretrained("rwcuffney/autotrain-pick_a_card-3726099225")
|
50 |
model = AutoModelForImageClassification.from_pretrained("rwcuffney/autotrain-pick_a_card-3726099225")
|