Update app.py

app.py

@@ -1,10 +1,13 @@
 import os
 from typing import Iterator
-
 import gradio as gr
-
 from model import run
 
+# Ensure the HF_TOKEN environment variable is set
+HF_TOKEN = os.environ.get("HF_TOKEN")
+if HF_TOKEN is None:
+    raise ValueError("Please set the HF_TOKEN environment variable.")
+
 HF_PUBLIC = os.environ.get("HF_PUBLIC", False)
 
 DEFAULT_SYSTEM_PROMPT = """\
@@ -272,4 +275,4 @@ with gr.Blocks(css='style.css') as demo:
         api_name=False,
     )
 
-demo.queue(max_size=32).launch(share=HF_PUBLIC)
+demo.queue(max_size=32).launch(share=HF_PUBLIC)