aifartist commited on
Commit
ce35e14
·
verified ·
1 Parent(s): 3a69d7a

Update gradio-app.py

Browse files
Files changed (1) hide show
  1. gradio-app.py +6 -1
gradio-app.py CHANGED
@@ -3,7 +3,12 @@ import gradio as gr
3
  from PIL import Image
4
  import torch
5
  from diffusers import DiffusionPipeline, AutoencoderTiny
6
- import os
 
 
 
 
 
7
 
8
  SAFETY_CHECKER = os.environ.get("SAFETY_CHECKER", None)
9
  TORCH_COMPILE = os.environ.get("TORCH_COMPILE", None)
 
3
  from PIL import Image
4
  import torch
5
  from diffusers import DiffusionPipeline, AutoencoderTiny
6
+ import os, sys
7
+
8
+ import subprocess
9
+
10
+
11
+ subprocess.check_call([sys.executable, "-m", "pip", "install", 'stable-fast'])
12
 
13
  SAFETY_CHECKER = os.environ.get("SAFETY_CHECKER", None)
14
  TORCH_COMPILE = os.environ.get("TORCH_COMPILE", None)