Spaces:
Running
Running
Ffftdtd5dtft
commited on
Update app.py
Browse files
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 |
-
|
58 |
-
|
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 |
-
|
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 |
-
|
225 |
-
|
226 |
-
|
|
|
|
|
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
|
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")
|