Update app.py

app.py

@@ -1,9 +1,12 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
+import sqlparse
+# from modelscope import snapshot_download
 
 # 加载模型和分词器
 model_name = "defog/llama-3-sqlcoder-8b"  # 使用更新的模型以提高性能
+# model_name = snapshot_download("stevie/llama-3-sqlcoder-8b")
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(
     model_name,