Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from transformers import AutoModel, AutoTokenizer, AutoModelForCausalLM
 from peft import PeftModel
 import re
+import os
 import akshare as ak
 import pandas as pd
 import random