second test

app.py

@@ -1,8 +1,21 @@
 import gradio as gr
 
-def greet(name):
-    return "Hello " + name + "!"
+from transformers import pipeline, AutoModelForImageClassification, AutoFeatureExtractor
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
+HF_MODEL_PATH = (
+    "ImageIN/convnext-base-224_finetuned_on_unlabelled_IA_with_snorkel_labels"
+)
 
-demo.launch()   
+classif_model = AutoModelForImageClassification.from_pretrained(HF_MODEL_PATH)
+feature_extractor = AutoFeatureExtractor.from_pretrained(HF_MODEL_PATH)
+
+classif_pipeline = pipeline(
+    "image-classification", model=classif_model, feature_extractor=feature_extractor
+)
+
+demo = gr.Interface(
+    fn=lambda x: classif_pipeline(x)[0]["label"],
+    inputs=gr.Image(type="pil"),
+    outputs="text",
+)
+demo.launch()