Update app.py

app.py

@@ -10,7 +10,7 @@ st.title("Classificador de Imagens com AutoTrain")
 model_name_or_path = "coan/designTrends"  # Substitua com seu modelo AutoTrain
 
 # Carregar o processador e o modelo
-@st.cache(allow_output_mutation=True)
+@st.cache_resource
 def load_model():
     processor = AutoProcessor.from_pretrained(model_name_or_path)
     model = AutoModelForImageClassification.from_pretrained(model_name_or_path)
@@ -41,4 +41,4 @@ if uploaded_file is not None:
         predicted_label = model.config.id2label[predicted_class_idx]
 
     # Exibir o resultado da previsão
-    st.write(f"Categoria prevista: **{predicted_label}**")
+    st.write(f"Categoria prevista: **{predicted_label}**")