Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM, set_seed
 from transformers import pipeline
-
+import os
 
 title = "Santa Model Generator"
 description = "Demo"