Update app.py

app.py

@@ -18,6 +18,7 @@ chat_history = []
 nltk.download('punkt_tab')
 token = os.environ.get('HF_TOKEN')
 spc = os.environ.get('SPC')
+spc = os.environ.get('SPC2')
 dataset_mng = DatasetManager(os.environ.get('DATASET'))
 tokeniz = AutoTokenizer.from_pretrained("iarfmoose/t5-base-question-generator")
 model = AutoModelForSeq2SeqLM.from_pretrained("iarfmoose/t5-base-question-generator")
@@ -357,4 +358,7 @@ with gr.Blocks() as demo:
         with gr.Tab("Chat"):
             gr.HTML(f"<iframe src='{spc}' width='100%' height='600'></iframe>")
 
+        with gr.Tab("Image Generation"):
+            gr.HTML(f"<iframe src='{spc2}' width='100%' height='600'></iframe>")
+
 demo.launch()