cocktailpeanut commited on
Commit
88f0b0b
·
1 Parent(s): 90c62ea
Files changed (1) hide show
  1. app.py +2 -2
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
- @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}")
 
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}")