Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -36,10 +36,9 @@ from huggingface_hub import from_pretrained_keras
|
|
36 |
|
37 |
# Cell 1: Image Classification Model
|
38 |
model1 = from_pretrained_keras("rocioadlc/EfficientNetV2L")
|
39 |
-
image_pipeline = pipeline(task="image-classification", model=model1)
|
40 |
|
41 |
def predict_image(input_img):
|
42 |
-
predictions =
|
43 |
return input_img, {p["label"]: p["score"] for p in predictions}
|
44 |
|
45 |
image_gradio_app = gr.Interface(
|
|
|
36 |
|
37 |
# Cell 1: Image Classification Model
|
38 |
model1 = from_pretrained_keras("rocioadlc/EfficientNetV2L")
|
|
|
39 |
|
40 |
def predict_image(input_img):
|
41 |
+
predictions = model1.predict(input_img)
|
42 |
return input_img, {p["label"]: p["score"] for p in predictions}
|
43 |
|
44 |
image_gradio_app = gr.Interface(
|