Spaces:
Running
Running
| import streamlit.web.cli as stcli | |
| import os, sys, socket | |
| # pip install protobuf==3.20.0 | |
| # pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117 nope | |
| # pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118 | |
| # pip install protobuf==3.20.0 | |
| def find_available_port(start_port, max_attempts=1000): | |
| port = start_port | |
| attempts = 0 | |
| while attempts < max_attempts: | |
| with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s: | |
| try: | |
| s.bind(("127.0.0.1", port)) | |
| # If successful, return the current port | |
| return port | |
| except socket.error: | |
| # If the port is in use, increment the port number and try again | |
| port += 1 | |
| attempts += 1 | |
| # Optional: Return None or raise an exception if no port is found within the attempts limit | |
| raise ValueError(f"Could not find an available port within {max_attempts} attempts starting from port {start_port}.") | |
| def resolve_path(path): | |
| resolved_path = os.path.abspath(os.path.join(os.getcwd(), path)) | |
| return resolved_path | |
| if __name__ == "__main__": | |
| dir_home = os.path.dirname(__file__) | |
| start_port = 8530 | |
| try: | |
| free_port = find_available_port(start_port) | |
| sys.argv = [ | |
| "streamlit", | |
| "run", | |
| resolve_path(os.path.join(os.path.dirname(__file__),"app.py")), | |
| # resolve_path(os.path.join(dir_home,"vouchervision", "VoucherVision_GUI.py")), | |
| "--global.developmentMode=false", | |
| # "--server.port=8545", | |
| f"--server.port={free_port}", | |
| f"--server.maxUploadSize=51200", | |
| # Toggle below for HF vs Local | |
| # "--is_hf=1", | |
| # "--is_hf=0", | |
| ] | |
| sys.exit(stcli.main()) | |
| except ValueError as e: | |
| print(e) |