RunningYou commited on
Commit
931a3ea
1 Parent(s): 73e49ee
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +1 -3
app.py CHANGED
@@ -1,16 +1,16 @@
1
  import imageio
2
  from PIL import Image
3
  import math
4
- import streamlit as st
5
  import numpy as np
6
  import torch
7
  import PIL
 
8
  import cv2
9
  import mediapipe as mp
10
  import gradio as gr
11
  from diffusers import StableDiffusionInpaintPipeline
12
 
13
- YOUR_TOKEN = st.secrets['USER_TOKEN']
14
 
15
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
16
  model_path = "runwayml/stable-diffusion-inpainting"
 
1
  import imageio
2
  from PIL import Image
3
  import math
 
4
  import numpy as np
5
  import torch
6
  import PIL
7
+ import os
8
  import cv2
9
  import mediapipe as mp
10
  import gradio as gr
11
  from diffusers import StableDiffusionInpaintPipeline
12
 
13
+ YOUR_TOKEN = os.environ.get("API_TOKEN") or True
14
 
15
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
16
  model_path = "runwayml/stable-diffusion-inpainting"
requirements.txt CHANGED
@@ -3,6 +3,4 @@ transformers
3
  mediapipe
4
  scipy
5
  ftfy
6
- imageio
7
- torch
8
- streamlit
 
3
  mediapipe
4
  scipy
5
  ftfy
6
+ torch