Jonny001 commited on
Commit
374cd21
·
verified ·
1 Parent(s): 706e59b

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -77
app.py DELETED
@@ -1,77 +0,0 @@
1
- import numpy as np
2
- import gradio as gr
3
- import roop.globals
4
- from roop.core import (
5
- start,
6
- decode_execution_providers,
7
- suggest_max_memory,
8
- suggest_execution_threads,
9
- )
10
- from roop.processors.frame.core import get_frame_processors_modules
11
- from roop.utilities import normalize_output_path
12
- import os
13
- from PIL import Image
14
-
15
-
16
- def swap_face(source_file, target_file, doFaceEnhancer):
17
-
18
- source_path = "input.jpg"
19
- target_path = "target.jpg"
20
-
21
- source_image = Image.fromarray(source_file)
22
- source_image.save(source_path)
23
- target_image = Image.fromarray(target_file)
24
- target_image.save(target_path)
25
-
26
- print("source_path: ", source_path)
27
- print("target_path: ", target_path)
28
-
29
- roop.globals.source_path = source_path
30
- roop.globals.target_path = target_path
31
- output_path = "output.jpg"
32
- roop.globals.output_path = normalize_output_path(
33
- roop.globals.source_path, roop.globals.target_path, output_path
34
- )
35
- if doFaceEnhancer:
36
- roop.globals.frame_processors = ["face_swapper", "face_enhancer"]
37
- else:
38
- roop.globals.frame_processors = ["face_swapper"]
39
- roop.globals.headless = True
40
- roop.globals.keep_fps = True
41
- roop.globals.keep_audio = True
42
- roop.globals.keep_frames = False
43
- roop.globals.many_faces = False
44
- roop.globals.video_encoder = "libx264"
45
- roop.globals.video_quality = 18
46
- roop.globals.max_memory = suggest_max_memory()
47
- roop.globals.execution_providers = decode_execution_providers(["cuda"])
48
- roop.globals.execution_threads = suggest_execution_threads()
49
-
50
- print(
51
- "start process",
52
- roop.globals.source_path,
53
- roop.globals.target_path,
54
- roop.globals.output_path,
55
- )
56
-
57
- for frame_processor in get_frame_processors_modules(
58
- roop.globals.frame_processors
59
- ):
60
- if not frame_processor.pre_check():
61
- return
62
-
63
- start()
64
- return output_path
65
-
66
- app = gr.Blocks()
67
-
68
- with app:
69
- gr.Interface(
70
- fn=swap_face,
71
- inputs=[gr.Image(), gr.Image(), gr.Checkbox(label="Apply", info="Face Enhancer")],
72
- outputs="image",
73
- theme="NoCrypt/miku",
74
- description="Sorry for the inconvenience. The model is currently running using CPU, which might affect performance. We appreciate your understanding."
75
- )
76
-
77
- app.launch()