Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -41,7 +41,7 @@ custom_title = "<span style='color: rgb(243, 239, 224);'>Green Greta</span>"
|
|
41 |
|
42 |
|
43 |
# Cell 1: Image Classification Model
|
44 |
-
image_pipeline = pipeline(task="image-classification", model="
|
45 |
|
46 |
def predict_image(input_img):
|
47 |
predictions = image_pipeline(input_img)
|
|
|
41 |
|
42 |
|
43 |
# Cell 1: Image Classification Model
|
44 |
+
image_pipeline = pipeline(task="image-classification", model="rocioadlc/EfficientNetV2L")
|
45 |
|
46 |
def predict_image(input_img):
|
47 |
predictions = image_pipeline(input_img)
|