Update app.py

app.py

@@ -1,15 +1,12 @@
-#!/usr/bin/env python
-# coding: utf-8
 
 # ### Keywords to Title Generator
 # - https://huggingface.co/EnglishVoice/t5-base-keywords-to-headline?text=diabetic+diet+plan
 # - Apache 2.0
 
-# In[7]:
-
 
 import torch
 from transformers import T5ForConditionalGeneration,T5Tokenizer
+import gradio as gr
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
@@ -17,11 +14,6 @@ model = T5ForConditionalGeneration.from_pretrained("EnglishVoice/t5-base-keyword
 tokenizer = T5Tokenizer.from_pretrained("EnglishVoice/t5-base-keywords-to-headline", clean_up_tokenization_spaces=True, legacy=False)
 model = model.to(device)
 
-
-
-# In[45]:
-
-
 def title_gen(keywords, diversity, temp):
 
  if keywords!= "":
@@ -59,14 +51,6 @@ def title_gen(keywords, diversity, temp):
     return titles
 
 
-# In[8]:
-
-
-import gradio as gr
-
-
-# In[48]:
-
 
 iface = gr.Interface(fn=title_gen, 
                      inputs=[gr.Textbox(label="Paste one or more keywords searated by a comma and hit 'Submit'.", lines=1), "checkbox", gr.Slider(0.1, 1.9, 1.2)],
@@ -85,54 +69,3 @@ iface = gr.Interface(fn=title_gen,
                     )
 
 iface.launch()
-
-
-# In[ ]:
-
-
-'''
-#Create a four button panel for changing parameters with one click
-
-def fn(text):
-    return ("Hello gradio!")
-
-with gr.Blocks () as demo:
-
-    with gr.Row(variant='compact') as PanelRow1: #first row: top
-        
-        with gr.Column(scale=0, min_width=180) as PanelCol5:
-            gr.HTML("")
-        with gr.Column(scale=0) as PanelCol4:
-            submit = gr.Button("Temp++", scale=0)
-        with gr.Column(scale=1) as PanelCol5:
-            gr.HTML("")
-        
-       
-    with gr.Row(variant='compact') as PanelRow2: #2nd row: left, right, middle
-
-        with gr.Column(min_width=100) as PanelCol1:
-            submit = gr.Button("Contrastive")
-        with gr.Column(min_width=100) as PanelCol2:
-            submit = gr.Button("Re-generate")
-        with gr.Column(min_width=100) as PanelCol3:
-            submit = gr.Button("Diversity Beam")
-        
-        with gr.Column(min_width=100) as PanelCol5:
-            gr.HTML("")
-        with gr.Column(min_width=100) as PanelCol5:
-            gr.HTML("")
-        with gr.Column(scale=0) as PanelCol5:
-            gr.HTML("")
-            
-    with gr.Row(variant='compact') as PanelRow3: #last row: down
-        with gr.Column(scale=0, min_width=180) as PanelCol7:
-            gr.HTML("")
-        with gr.Column(scale=1) as PanelCol6:
-            submit = gr.Button("Temp--", scale=0)
-            
-        with gr.Column(scale=0) as PanelCol5:
-            gr.HTML("")
-
-demo.launch()
-'''
-