Spaces:
Runtime error
Runtime error
Commit
·
88f0b0b
1
Parent(s):
90c62ea
update
Browse files
app.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
import gradio as gr
|
2 |
import torch
|
3 |
-
import spaces
|
4 |
from PIL import Image
|
5 |
import os
|
6 |
from transformers import CLIPTokenizer, CLIPTextModel, AutoProcessor, T5EncoderModel, T5TokenizerFast
|
@@ -233,7 +233,7 @@ def compute_text_embeddings(prompt=""):
|
|
233 |
pooled_prompt_embeds = prompt_embeds.pooler_output
|
234 |
return pooled_prompt_embeds
|
235 |
|
236 |
-
|
237 |
def generate(input_image, prompt="", guidance_scale=3.5, num_inference_steps=28, num_images=2, seed=None, aspect_ratio="1:1", progress=gr.Progress(track_tqdm=True)):
|
238 |
try:
|
239 |
logger.info(f"Starting generation with prompt: {prompt}")
|
|
|
1 |
import gradio as gr
|
2 |
import torch
|
3 |
+
#import spaces
|
4 |
from PIL import Image
|
5 |
import os
|
6 |
from transformers import CLIPTokenizer, CLIPTextModel, AutoProcessor, T5EncoderModel, T5TokenizerFast
|
|
|
233 |
pooled_prompt_embeds = prompt_embeds.pooler_output
|
234 |
return pooled_prompt_embeds
|
235 |
|
236 |
+
#@spaces.GPU(duration=75)
|
237 |
def generate(input_image, prompt="", guidance_scale=3.5, num_inference_steps=28, num_images=2, seed=None, aspect_ratio="1:1", progress=gr.Progress(track_tqdm=True)):
|
238 |
try:
|
239 |
logger.info(f"Starting generation with prompt: {prompt}")
|