Merge branch 'main' of https://huggingface.co/spaces/ALVHB95/TFM_DataScience_APP

app.py

@@ -32,14 +32,13 @@ 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
-model1 = from_pretrained_keras("rocioadlc/EfficientNetV2L")
-image_pipeline = pipeline(task="image-classification", model=model1)
+image_pipeline = pipeline(task="image-classification", model="microsoft/resnet-50")
 
 def predict_image(input_img):
-    predictions = image_pipeline(input_img)
+    predictions = model1.predict(input_img)
     return input_img, {p["label"]: p["score"] for p in predictions} 
 
 image_gradio_app = gr.Interface(