Spaces:
Running
on
Zero
Running
on
Zero
prev commit (#4)
Browse files- prev commit (410a20ae5d3cf2beb4ac08572e0a3c84d20fbf9c)
app.py
CHANGED
@@ -9,7 +9,6 @@ from typing import Union, List, Optional
|
|
9 |
import numpy as np
|
10 |
import time
|
11 |
import zipfile
|
12 |
-
import re
|
13 |
|
14 |
# Description for the app
|
15 |
DESCRIPTION = """## Qwen Image Hpc/."""
|
@@ -61,16 +60,6 @@ def get_duration(
|
|
61 |
per_image_duration = 5 + num_inference_steps * step_duration
|
62 |
return 10 + num_images * per_image_duration
|
63 |
|
64 |
-
# Positive Magic texts
|
65 |
-
positive_magic = {
|
66 |
-
"en": "Ultra HD, 4K, cinematic composition.",
|
67 |
-
"zh": "超清,4K,电影级构图"
|
68 |
-
}
|
69 |
-
|
70 |
-
# Function to check if the prompt contains Chinese characters
|
71 |
-
def contains_chinese(text):
|
72 |
-
return bool(re.search(r'[\u4e00-\u9fff]', text))
|
73 |
-
|
74 |
# Generation function for Qwen/Qwen-Image
|
75 |
def generate_qwen(
|
76 |
prompt: str,
|
@@ -133,15 +122,9 @@ def generate(
|
|
133 |
zip_images: bool,
|
134 |
progress=gr.Progress(track_tqdm=True),
|
135 |
):
|
136 |
-
# Enhance the prompt with Positive Magic based on language
|
137 |
-
if contains_chinese(prompt):
|
138 |
-
final_prompt = prompt + " " + positive_magic["zh"]
|
139 |
-
else:
|
140 |
-
final_prompt = prompt + " " + positive_magic["en"]
|
141 |
-
|
142 |
final_negative_prompt = negative_prompt if use_negative_prompt else ""
|
143 |
return generate_qwen(
|
144 |
-
prompt=
|
145 |
negative_prompt=final_negative_prompt,
|
146 |
seed=seed,
|
147 |
width=width,
|
|
|
9 |
import numpy as np
|
10 |
import time
|
11 |
import zipfile
|
|
|
12 |
|
13 |
# Description for the app
|
14 |
DESCRIPTION = """## Qwen Image Hpc/."""
|
|
|
60 |
per_image_duration = 5 + num_inference_steps * step_duration
|
61 |
return 10 + num_images * per_image_duration
|
62 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
63 |
# Generation function for Qwen/Qwen-Image
|
64 |
def generate_qwen(
|
65 |
prompt: str,
|
|
|
122 |
zip_images: bool,
|
123 |
progress=gr.Progress(track_tqdm=True),
|
124 |
):
|
|
|
|
|
|
|
|
|
|
|
|
|
125 |
final_negative_prompt = negative_prompt if use_negative_prompt else ""
|
126 |
return generate_qwen(
|
127 |
+
prompt=prompt,
|
128 |
negative_prompt=final_negative_prompt,
|
129 |
seed=seed,
|
130 |
width=width,
|