Ffftdtd5dtft commited on
Commit
c95cc1d
·
verified ·
1 Parent(s): ae48414

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -8
app.py CHANGED
@@ -54,9 +54,8 @@ def save_object_to_redis(key, obj):
54
  redis_client = connect_to_redis()
55
  redis_client = reconnect_if_needed(redis_client)
56
  try:
57
- if not redis_client.exists(key): # Solo guarda si no existe
58
- redis_client.set(key, pickle.dumps(obj))
59
- print(f"Object saved to Redis: {key}")
60
  except redis.exceptions.RedisError as e:
61
  print(f"Failed to save object to Redis: {e}")
62
 
@@ -71,7 +70,7 @@ def get_model_or_download(model_id, redis_key, loader_func):
71
  print(f"Model downloaded and saved to Redis: {redis_key}")
72
  except Exception as e:
73
  print(f"Failed to load or save model: {e}")
74
- return model
75
 
76
  def generate_image(prompt):
77
  redis_key = f"generated_image_{prompt}"
@@ -221,13 +220,16 @@ for _ in range(num_processes):
221
  processes.append(p)
222
 
223
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
224
- text_to_image_pipeline = get_model_or_download("CompVis/stable-diffusion-v1-4", "text_to_image_model", StableDiffusionPipeline.from_pretrained).to(device)
225
- img2img_pipeline = get_model_or_download("CompVis/stable-diffusion-v1-4", "img2img_model", StableDiffusionImg2ImgPipeline.from_pretrained).to(device)
226
- flux_pipeline = get_model_or_download("CompVis/stable-diffusion-flux", "flux_model", FluxPipeline.from_pretrained).to(device)
 
 
227
  text_gen_pipeline = transformers_pipeline("text-generation", model="bigcode/starcoder", tokenizer="bigcode/starcoder", device=0)
228
  music_gen = load_object_from_redis("music_gen") or MusicGen.from_pretrained('melody')
229
- meta_llama_pipeline = get_model_or_download("meta/meta-llama-7b", "meta_llama_model", transformers_pipeline)
230
 
 
231
  gen_image_tab = gr.Interface(generate_image, gr.inputs.Textbox(label="Prompt:"), gr.outputs.Image(type="pil"), title="Generate Image")
232
  edit_image_tab = gr.Interface(edit_image_with_prompt, [gr.inputs.Image(type="pil", label="Image:"), gr.inputs.Textbox(label="Prompt:"), gr.inputs.Slider(0.1, 1.0, 0.75, step=0.05, label="Strength:")], gr.outputs.Image(type="pil"), title="Edit Image")
233
  generate_song_tab = gr.Interface(generate_song, [gr.inputs.Textbox(label="Prompt:"), gr.inputs.Slider(5, 60, 10, step=1, label="Duration (s):")], gr.outputs.Audio(type="numpy"), title="Generate Songs")
 
54
  redis_client = connect_to_redis()
55
  redis_client = reconnect_if_needed(redis_client)
56
  try:
57
+ redis_client.set(key, pickle.dumps(obj))
58
+ print(f"Object saved to Redis: {key}")
 
59
  except redis.exceptions.RedisError as e:
60
  print(f"Failed to save object to Redis: {e}")
61
 
 
70
  print(f"Model downloaded and saved to Redis: {redis_key}")
71
  except Exception as e:
72
  print(f"Failed to load or save model: {e}")
73
+ return None
74
 
75
  def generate_image(prompt):
76
  redis_key = f"generated_image_{prompt}"
 
220
  processes.append(p)
221
 
222
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
223
+
224
+ # Cargar modelos
225
+ text_to_image_pipeline = get_model_or_download("CompVis/stable-diffusion-v1-4", "text_to_image_model", StableDiffusionPipeline.from_pretrained)
226
+ img2img_pipeline = get_model_or_download("CompVis/stable-diffusion-v1-4", "img2img_model", StableDiffusionImg2ImgPipeline.from_pretrained)
227
+ flux_pipeline = get_model_or_download("black-forest-labs/FLUX.1-schnell", "flux_model", FluxPipeline.from_pretrained)
228
  text_gen_pipeline = transformers_pipeline("text-generation", model="bigcode/starcoder", tokenizer="bigcode/starcoder", device=0)
229
  music_gen = load_object_from_redis("music_gen") or MusicGen.from_pretrained('melody')
230
+ meta_llama_pipeline = get_model_or_download("meta-llama/Meta-Llama-3.1-8B-Instruct", "meta_llama_model", transformers_pipeline)
231
 
232
+ # Definir interfaces de usuario
233
  gen_image_tab = gr.Interface(generate_image, gr.inputs.Textbox(label="Prompt:"), gr.outputs.Image(type="pil"), title="Generate Image")
234
  edit_image_tab = gr.Interface(edit_image_with_prompt, [gr.inputs.Image(type="pil", label="Image:"), gr.inputs.Textbox(label="Prompt:"), gr.inputs.Slider(0.1, 1.0, 0.75, step=0.05, label="Strength:")], gr.outputs.Image(type="pil"), title="Edit Image")
235
  generate_song_tab = gr.Interface(generate_song, [gr.inputs.Textbox(label="Prompt:"), gr.inputs.Slider(5, 60, 10, step=1, label="Duration (s):")], gr.outputs.Audio(type="numpy"), title="Generate Songs")