Jonny001 commited on
Commit
b7248bb
·
verified ·
1 Parent(s): 4497450

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -78
app.py DELETED
@@ -1,78 +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
-
67
- app = gr.Blocks()
68
-
69
- with app:
70
- gr.Interface(
71
- fn=swap_face,
72
- inputs=[gr.Image(), gr.Image(), gr.Checkbox(label="Apply Face Enhancer")],
73
- outputs="image",
74
- description="Sorry for the inconvenience. The model is currently running on the CPU, which might affect performance. We appreciate your understanding. ( Apply Face Enhancer to get better result. )",
75
- theme="Nymbo/Nymbo_Theme"
76
- )
77
-
78
- app.launch()