Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch.nn.functional as F
+import torch.nn as nn
+import re
 
 model_path = "ssocean/NAIP"  # 更换为你的模型路径
 model = AutoModelForSequenceClassification.from_pretrained(model_path, num_labels=1, load_in_8bit=True)