Update app.py

app.py

@@ -1,5 +1,15 @@
-from transformers import AutoModelForMaskedLM, AutoTokenizer
 from huggingface_hub import notebook_login
+import torch
+import transformers
+from transformers import AutoModelForMaskedLM, AutoTokenizer
+from transformers import pipeline
+from transformers import LlamaTokenizer, LlamaForCausalLM
+import time
+import csv
+import locale
+locale.getpreferredencoding = lambda: "UTF-8"
+
+
 notebook_login()
 checkpoint = "ziqingyang/chinese-alpaca-2-7b"
 tokenizer = AutoTokenizer.from_pretrained(checkpoint)
@@ -28,4 +38,4 @@ model = AutoModelForMaskedLM.from_pretrained(checkpoint)
 
 
 model.push_to_hub("gutalk_classification") 
-tokenizer.push_to_hub("gutalk_classification")
+tokenizer.push_to_hub("gutalk_classification")