Upload 4 files

app.py

@@ -1,17 +1,17 @@
 import streamlit as st
 from PIL import Image
 import torch
-from transformers import AutoFeatureExtractor, AutoModelForImageClassification
+from transformers import AutoImageProcessor, ConvNextForImageClassification
 import pandas as pd
 from utils.matcher import fuzzy_match
 
 @st.cache_resource
 def load_model():
-    extractor = AutoFeatureExtractor.from_pretrained("nateraw/food-classification")
-    model = AutoModelForImageClassification.from_pretrained("nateraw/food-classification")
-    return extractor, model
+    processor = AutoImageProcessor.from_pretrained("duongna/convnext-tiny-food101")
+    model = ConvNextForImageClassification.from_pretrained("duongna/convnext-tiny-food101")
+    return processor, model
 
-extractor, model = load_model()
+processor, model = load_model()
 data = pd.read_csv("kaloriedata.csv")
 madliste = data["navn"].tolist()
 
@@ -23,7 +23,7 @@ if uploaded:
     img = Image.open(uploaded).convert("RGB")
     st.image(img, caption="Uploadet billede", use_container_width=True)
 
-    inputs = extractor(images=img, return_tensors="pt")
+    inputs = processor(images=img, return_tensors="pt")
     with torch.no_grad():
         logits = model(**inputs).logits
     probs = torch.nn.functional.softmax(logits, dim=-1)