Nuno-Tome commited on
Commit
576a32a
1 Parent(s): fee4ed8

no message

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -73,10 +73,10 @@ def main():
73
  st.write("# FLAG 3")
74
 
75
  #modle instance
76
- classifier_pipeline = pipeline('image-classification', model=chosen_model_name, device=0)
77
  #classifier_pipeline = pipeline('image-classification', model="nateraw/vit-age-classifier", device=0)
78
  st.write("# FLAG 4")
79
-
80
  #classification
81
  classification_result = classify(image_object, classifier_pipeline)
82
  st.write(classification_result)
 
73
  st.write("# FLAG 3")
74
 
75
  #modle instance
76
+ classifier_pipeline = pipeline('image-classification', model=chosen_model_name)
77
  #classifier_pipeline = pipeline('image-classification', model="nateraw/vit-age-classifier", device=0)
78
  st.write("# FLAG 4")
79
+
80
  #classification
81
  classification_result = classify(image_object, classifier_pipeline)
82
  st.write(classification_result)