Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import tensorflow as tf
 from huggingface_hub import login, create_repo, upload_file
+import os
 from transformers import AutoTokenizer, TFAutoModelForCausalLM
 policy = tf.keras.mixed_precision.Policy('mixed_bfloat16')
 tf.keras.mixed_precision.set_global_policy(policy)