JoPmt commited on
Commit
a13b293
1 Parent(s): 4c7fffb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -3,13 +3,15 @@ import math
3
  import time
4
  import numpy
5
  import random
 
6
  import threading
7
  import gradio as gr
8
  from PIL import Image, ImageOps
9
  from moviepy import VideoFileClip
10
  from datetime import datetime, timedelta
11
- from huggingface_hub import hf_hub_download, snapshot_download
12
-
 
13
  import insightface
14
  from insightface.app import FaceAnalysis
15
  from facexlib.parsing import init_parsing_model
@@ -331,7 +333,7 @@ with gr.Blocks() as demo:
331
  latents, seed = infer(
332
  prompt,
333
  image_input,
334
- num_inference_steps=50,
335
  guidance_scale=7.0,
336
  seed=seed_value,
337
  progress=progress,
 
3
  import time
4
  import numpy
5
  import random
6
+ import spaces
7
  import threading
8
  import gradio as gr
9
  from PIL import Image, ImageOps
10
  from moviepy import VideoFileClip
11
  from datetime import datetime, timedelta
12
+ from huggingface_hub import hf_hub_download, snapshot_download, login
13
+ HF_TOKEN=os.environ.get('HF_TOKEN')
14
+ login(token=HF_TOKEN)
15
  import insightface
16
  from insightface.app import FaceAnalysis
17
  from facexlib.parsing import init_parsing_model
 
333
  latents, seed = infer(
334
  prompt,
335
  image_input,
336
+ num_inference_steps=5,
337
  guidance_scale=7.0,
338
  seed=seed_value,
339
  progress=progress,