app21

app.py

@@ -32,10 +32,11 @@ from langchain_community.document_loaders import WebBaseLoader
 from pydantic import BaseModel
 import shutil
 
-
+from huggingface_hub import from_pretrained_keras
 
 # Cell 1: Image Classification Model
-image_pipeline = pipeline(task="image-classification", model="microsoft/resnet-50")
+model1 = from_pretrained_keras("rocioadlc/EfficientNetV2L")
+image_pipeline = pipeline(task="image-classification", model=model1)
 
 def predict_image(input_img):
     predictions = image_pipeline(input_img)