Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 from transformers import pipeline
 from datasets import load_dataset
 
-pipeline = pipeline(task="image-classification", model="puqi/1")
+pipeline = pipeline(model="puqi/1")
 data_files = {"train": "train_input_small.npy", "val": "val_input_small.npy"}
 dataset = load_dataset("puqi/test ", data_files=data_files)