Update app.py

app.py

@@ -9,6 +9,7 @@ from time import sleep
 import tiktoken
 # required for HF SPACES
 from huggingface_hub import hf_hub_download
+import os
 
 # for counting the tokens in the prompt and in the result
 #context_count = len(encoding.encode(yourtext))