islasher commited on
Commit
6e92121
·
verified ·
1 Parent(s): e46b5cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -16,11 +16,11 @@ def pipeline():
16
  # 1. Scrape news
17
  headlines = scan_news()
18
 
19
- # 2. LLM summary
20
- summary = summarize_headlines(headlines)
21
 
22
  # 3. LLM image prompt
23
- img_prompt = generate_image_prompt(summary)
24
 
25
  # 4. Generate image
26
  # img = generate_image(img_prompt)
@@ -33,16 +33,16 @@ def pipeline():
33
 
34
  with gr.Blocks(title="AI News Meme Generator") as app:
35
  gr.Markdown("# News Meme Generator")
36
- gr.Markdown("Este modelo lee noticias, las resume, genera un prompt, crea una imagen y hace un meme.")
37
 
38
  btn = gr.Button("Generate Meme")
39
 
40
- out_headlines = gr.JSON(label="Titulares")
41
- out_summary = gr.Textbox(label="Resumen Humorístico")
42
- out_prompt = gr.Textbox(label="Prompt para Imagen")
43
  out_image = gr.Image(label="Meme Final")
44
 
45
- btn.click(fn=pipeline, outputs=[out_headlines, out_summary, out_prompt])#, out_image])
46
 
47
  app.launch(ssr_mode=False, share=False)
48
 
 
16
  # 1. Scrape news
17
  headlines = scan_news()
18
 
19
+ # # 2. LLM summary
20
+ # summary = summarize_headlines(headlines)
21
 
22
  # 3. LLM image prompt
23
+ img_prompt = generate_image_prompt(headlines)#summary)
24
 
25
  # 4. Generate image
26
  # img = generate_image(img_prompt)
 
33
 
34
  with gr.Blocks(title="AI News Meme Generator") as app:
35
  gr.Markdown("# News Meme Generator")
36
+ gr.Markdown("Este modelo lee un titular, genera un prompt, crea una imagen y hace un meme.")
37
 
38
  btn = gr.Button("Generate Meme")
39
 
40
+ out_headlines = gr.JSON(label="Headline")
41
+ # out_summary = gr.Textbox(label="Resumen Humorístico")
42
+ out_prompt = gr.Textbox(label="Prompt for Image")
43
  out_image = gr.Image(label="Meme Final")
44
 
45
+ btn.click(fn=pipeline, outputs=[out_headlines, out_prompt])#, out_image]) #out_summary
46
 
47
  app.launch(ssr_mode=False, share=False)
48