Upload app.py

app.py

@@ -19,7 +19,7 @@ model = model.to(device)
 
 
 
-# In[37]:
+# In[45]:
 
 
 def title_gen(keywords, diversity, temp):
@@ -50,11 +50,11 @@ def title_gen(keywords, diversity, temp):
         #top_p = 0.60,
     )
 
-    titles = ""
+    titles = "<p align=center>Title Suggestions:</p>"
     
     for i in range(len(beam_outputs)):
         result = tokenizer.decode(beam_outputs[i], skip_special_tokens=True)
-        titles += f"<p align=center><b>{result}</b></p>" #Create string with titles and <br> tag for html reading in gradio html
+        titles += f"<p align=center><b>{result}</b></p></p>" #Create string with titles and <br> tag for html reading in gradio html
     
     return titles
 
@@ -65,7 +65,7 @@ def title_gen(keywords, diversity, temp):
 import gradio as gr
 
 
-# In[40]:
+# In[48]:
 
 
 iface = gr.Interface(fn=title_gen, 
@@ -74,7 +74,14 @@ iface = gr.Interface(fn=title_gen,
                      title="AI Keywords to Title Generator", 
                      #description="Turn keywords into creative suggestions",
                      article="<div align=left><h1>AI Creative Title Generator</h1><li>With just keywords, generate a list of creative titles.</li><li>Click on Submit to generate more title options.</li><li>Tweak slider for less or more creative titles</li><li>Check 'diversity' to turn on diversity beam search</li><p>AI Model:<br><li>T5 Model trained on a dataset of titles and related keywords</li><li>Original model id: EnglishVoice/t5-base-keywords-to-headline by English Voice AI Labs</li></p><p>Default parameter details:<br><li>Temperature = 1.2, no_repeat_ngram_size=3, top_k = 15, penalty_alpha = 0.8, max_new_tokens = 30</li><p>Diversity beam search params:<br><li>num_beams=20, diversity_penalty=0.8, num_beam_groups=20</li></div>",
-                     flagging_mode='never'
+                     flagging_mode='never',
+                     examples=[
+                                ["new, weight loss, lifestyle"],
+                                ["launch, free, dating, app"],
+                                ["AI, text to video, app"],
+                                ["new movie, watch, free streaming"],
+                              ],
+                     cache_examples=True,
                     )
 
 iface.launch()