Update app.py

app.py

@@ -1,7 +1,11 @@
 import os
+import json
 import torch
+import shutil
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, TextIteratorStreamer
 from transformers.generation import LogitsProcessor
+import huggingface_hub
+from huggingface_hub import Repository
 from threading import Thread
 import gradio as gr
 
@@ -44,6 +48,9 @@ LANGUAGE_IDS = {
 STATS_REPO = "https://huggingface.co/datasets/ltg/usage_statistics"
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
+dataset = Repository(
+    local_dir="data", clone_from=STATS_REPO, use_auth_token=HF_TOKEN
+)
 
 # log the timestamp of the query
 def add_anonymous_usage_log(path):
@@ -57,7 +64,7 @@ def add_anonymous_usage_log(path):
     except:
         shutil.rmtree("data")
         dataset = Repository(
-            local_dir="data", clone_from=LOGS_REPO, use_auth_token=HF_TOKEN
+            local_dir="data", clone_from=STATS_REPO, use_auth_token=HF_TOKEN
         )
         with open(path, "a") as f:
             line = json.dumps(str(datetime.now()), ensure_ascii=False)