Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch.nn.functional as F
 
 model_path = "ssocean/NAIP"  # 更换为你的模型路径
-model = AutoModelForSequenceClassification.from_pretrained(model_path, num_labels=1, load_in_8bit=True)
+model = AutoModelForSequenceClassification.from_pretrained(model_path, num_labels=1,)# load_in_8bit=True
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 
 device = "cuda" if torch.cuda.is_available() else "cpu"