Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -27,7 +27,7 @@ qwen_model = Qwen2VLForConditionalGeneration.from_pretrained(
|
|
27 |
qwen_processor = AutoProcessor.from_pretrained("prithivMLmods/JSONify-Flux", trust_remote_code=True)
|
28 |
|
29 |
# Prompt Enhancer
|
30 |
-
enhancer_long = pipeline("summarization", model="
|
31 |
|
32 |
MAX_SEED = np.iinfo(np.int32).max
|
33 |
MAX_IMAGE_SIZE = 1024 # Reduced to prevent memory issues
|
@@ -140,7 +140,7 @@ title = """<h1 align="center">FLUX.1-dev with Qwen2VL Captioner and Prompt Enhan
|
|
140 |
<p><center>
|
141 |
<a href="https://huggingface.co/black-forest-labs/FLUX.1-dev" target="_blank">[FLUX.1-dev Model]</a>
|
142 |
<a href="https://huggingface.co/prithivMLmods/JSONify-Flux" target="_blank">[JSONify Flux Model]</a>
|
143 |
-
<a href="https://huggingface.co/
|
144 |
<p align="center">Create long prompts from images or enhance your short prompts with prompt enhancer</p>
|
145 |
</center></p>
|
146 |
"""
|
|
|
27 |
qwen_processor = AutoProcessor.from_pretrained("prithivMLmods/JSONify-Flux", trust_remote_code=True)
|
28 |
|
29 |
# Prompt Enhancer
|
30 |
+
enhancer_long = pipeline("summarization", model="prithivMLmods/t5-Flan-Prompt-Enhance", device=device)
|
31 |
|
32 |
MAX_SEED = np.iinfo(np.int32).max
|
33 |
MAX_IMAGE_SIZE = 1024 # Reduced to prevent memory issues
|
|
|
140 |
<p><center>
|
141 |
<a href="https://huggingface.co/black-forest-labs/FLUX.1-dev" target="_blank">[FLUX.1-dev Model]</a>
|
142 |
<a href="https://huggingface.co/prithivMLmods/JSONify-Flux" target="_blank">[JSONify Flux Model]</a>
|
143 |
+
<a href="https://huggingface.co/prithivMLmods/t5-Flan-Prompt-Enhance" target="_blank">[Prompt Enhancer t5]</a>
|
144 |
<p align="center">Create long prompts from images or enhance your short prompts with prompt enhancer</p>
|
145 |
</center></p>
|
146 |
"""
|