KyanChen commited on
Commit
d89338f
1 Parent(s): f549064

Upload 3 files

Browse files
Files changed (2) hide show
  1. main_page.py +1 -0
  2. requirements.txt +3 -4
main_page.py CHANGED
@@ -1,5 +1,6 @@
1
  import streamlit as st
2
  import os
 
3
  # https://huggingface.co/spaces/KyanChen/ai-photo-gallery
4
 
5
  st.set_page_config(
 
1
  import streamlit as st
2
  import os
3
+ os.system("mim install mmcv>=2.0.0rc4")
4
  # https://huggingface.co/spaces/KyanChen/ai-photo-gallery
5
 
6
  st.set_page_config(
requirements.txt CHANGED
@@ -1,8 +1,7 @@
1
- opencv-python_headless
2
- torch --extra-index-url https://download.pytorch.org/whl/cpu
3
- torchvision --extra-index-url https://download.pytorch.org/whl/cpu
4
  openmim
5
- mmcv>=2.0.0rc4 -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.13/index.html
6
  streamlit
7
  watchdog
8
  terminaltables
 
1
+ torch
2
+ torchvision
 
3
  openmim
4
+ mmcv>=2.0.0rc4
5
  streamlit
6
  watchdog
7
  terminaltables