Paper99 commited on
Commit
4019c56
1 Parent(s): 23336f1

Update README

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. app.py +3 -2
README.md CHANGED
@@ -1,6 +1,6 @@
1
  ---
2
  title: PhotoMaker
3
- emoji: 📚
4
  colorFrom: red
5
  colorTo: gray
6
  sdk: gradio
 
1
  ---
2
  title: PhotoMaker
3
+ emoji: 📷
4
  colorFrom: red
5
  colorTo: gray
6
  sdk: gradio
app.py CHANGED
@@ -15,7 +15,8 @@ from style_template import styles
15
 
16
  # global variable
17
  base_model_path = 'SG161222/RealVisXL_V3.0'
18
- device = "cuda"
 
19
  MAX_SEED = np.iinfo(np.int32).max
20
  STYLE_NAMES = list(styles.keys())
21
  DEFAULT_STYLE_NAME = "Photographic (Default)"
@@ -276,4 +277,4 @@ with gr.Blocks(css=css) as demo:
276
 
277
  gr.Markdown(article)
278
 
279
- demo.launch()
 
15
 
16
  # global variable
17
  base_model_path = 'SG161222/RealVisXL_V3.0'
18
+
19
+ device = "cuda" if torch.cuda.is_available() else "cpu"
20
  MAX_SEED = np.iinfo(np.int32).max
21
  STYLE_NAMES = list(styles.keys())
22
  DEFAULT_STYLE_NAME = "Photographic (Default)"
 
277
 
278
  gr.Markdown(article)
279
 
280
+ demo.launch()