Shot2Story / app.py
mhan's picture
Update app.py
4246f82 verified
import os
import random
import streamlit as st
import streamlit.components.v1 as components
st.set_page_config(layout="wide")
# Message for users
st.markdown("""
**Notice**: If you encounter any issues opening the demo or using the chatbot, try reloading the page which would reschedule a new demo instance.
Please report the issues [here](https://huggingface.co/spaces/mhan/Shot2Story/discussions/1).
We will respond ASAP.
""")
# List of backend URLs
backends = ["backend1", "backend2"]
# Randomly select a backend URL
backend = random.choice(backends)
backend_url = os.getenv(backend, "https://www.bytedance.com/en/") # 如果环境变量不存在,则使用默认 URL
iframe_html = f'<iframe src="{backend_url}" width="100%" height="800"></iframe>'
components.html(iframe_html, height=800)