From b860d5b90ac4e7cd862f7d00c2018419603fc2ce Mon Sep 17 00:00:00 2001
From: Baize <project-baize@users.noreply.huggingface.co>
Date: Tue, 4 Apr 2023 06:42:54 +0000
Subject: [PATCH] Update app.py

---
 app.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/app.py b/app.py
index 9f98474..d55d69b 100644
--- a/app.py
+++ b/app.py
@@ -44,6 +44,7 @@ def predict(text,
         begin_length = len(prompt)
     torch.cuda.empty_cache()
     input_ids = inputs["input_ids"].to(device)
+    global total_count
     total_count += 1
     print(total_count)
     with torch.no_grad():