Upload app.py

app.py

@@ -37,6 +37,21 @@ model = LlamaForCausalLM.from_pretrained(
     device_map='auto',
 )
 
+
+DATASET_REPO_URL = "https://huggingface.co/datasets/Claimed/logger"
+DATA_FILENAME = "data.csv"
+DATA_FILE = os.path.join("data", DATA_FILENAME)
+
+def store_log():
+    with open(DATA_FILE, "a") as csvfile:
+        writer = csv.DictWriter(csvfile, fieldnames=["count", "time"])
+        writer.writerow(
+            {"count": 1, "time": str(datetime.now())}
+        )
+    commit_url = repo.push_to_hub()
+    print(commit_url)
+    return 
+
 ########## DEFINING FUNCTIONS ###################
 
 def mean_pooling(model_output, attention_mask):
@@ -189,6 +204,8 @@ def classifier(userin, SearchType):
 
     md_class = '\n'.join(class_links)
 
+    store_log()
+
     return md_class
 
 def generateresponse(history, temp, top_p, tokens):
@@ -219,10 +236,7 @@ def generateresponse(history, temp, top_p, tokens):
     
     response = f"Response{outputs}"
 
-    f = open(os.path.join(os.path.dirname(__file__), "Model_bert/log.log"), "a")
-    count = 1
-    f.write(f'{count}\n')
-    f.close()
+    store_log()
     
     return response
 
@@ -254,10 +268,7 @@ def run_model(userin, dropd, temp, top_p, tokens):
     
     response = f"Response{outputs}"
 
-    f = open(os.path.join(os.path.dirname(__file__), "Model_bert/log.log"), "a")
-    count = 1
-    f.write(f'{count}\n')
-    f.close()
+    store_log()
     
     return response
 
@@ -293,10 +304,7 @@ def prosecute(application, priorart, temp, top_p, tokens):
     
     response = f"Response{outputs}"
 
-    f = open(os.path.join(os.path.dirname(__file__), "Model_bert/log.log"), "a")
-    count = 1
-    f.write(f'{count}\n')
-    f.close()
+    store_log()
     
     return response
 
@@ -329,10 +337,7 @@ def ideator(userin, temp, top_p, tokens):
     
     response = f"Response{outputs}"
 
-    f = open(os.path.join(os.path.dirname(__file__), "Model_bert/log.log"), "a")
-    count = 1
-    f.write(f'{count}\n')
-    f.close()
+    store_log()
     
     return response
 
@@ -364,10 +369,7 @@ def Chat(userin, temp, top_p, tokens):
     
     response = f"Response{outputs}"
 
-    f = open(os.path.join(os.path.dirname(__file__), "Model_bert/log.log"), "a")
-    count = 1
-    f.write(f'{count}\n')
-    f.close()
+    store_log()
     
     return response
        
@@ -563,3 +565,4 @@ with gr.Blocks(title='Claimed', theme=theme) as demo:
     """)
 demo.queue(max_size=15)
 demo.launch()
+