Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -32,10 +32,11 @@ from langchain_community.document_loaders import WebBaseLoader
|
|
32 |
from pydantic import BaseModel
|
33 |
import shutil
|
34 |
|
35 |
-
|
36 |
|
37 |
# Cell 1: Image Classification Model
|
38 |
-
|
|
|
39 |
|
40 |
def predict_image(input_img):
|
41 |
predictions = image_pipeline(input_img)
|
|
|
32 |
from pydantic import BaseModel
|
33 |
import shutil
|
34 |
|
35 |
+
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 = image_pipeline(input_img)
|