Update app.py

app.py

@@ -7,16 +7,19 @@ import gradio as gr
 model = EncoderDecoderModel.from_pretrained("imamnurby/rob2rand_chen_w_prefix_c_fc")
 tokenizer = RobertaTokenizer.from_pretrained("imamnurby/rob2rand_chen_w_prefix_c_fc")
 
-def generate_preds(gen_mode, desc):
-    desc = desc.lower()
+from transformers import RobertaTokenizer, EncoderDecoderModel
+import pandas as pd
+def generate_taps(gen_mode, desc):
+    input_desc = desc.lower()
     if gen_mode=="Channel":
-        desc = "GENERATE TRIGGER AND ACTION CHANNEL ONLY <pf> " + desc
+        input_desc = "GENERATE TRIGGER AND ACTION CHANNEL ONLY <pf> " + input_desc 
     elif gen_mode=="Function":
-        desc = "GENERATE BOTH CHANNEL AND FUNCTION FOR TRIGGER AND ACTION <pf> " + desc
+        input_desc = "GENERATE BOTH CHANNEL AND FUNCTION FOR TRIGGER AND ACTION <pf> " + input_desc 
     
+    model = EncoderDecoderModel.from_pretrained("imamnurby/rob2rand_chen_w_prefix_c_fc")
+    tokenizer = RobertaTokenizer.from_pretrained("imamnurby/rob2rand_chen_w_prefix_c_fc")
     
-    
-    input_ids = tokenizer.encode(desc, return_tensors='pt')
+    input_ids = tokenizer.encode(input_desc, return_tensors='pt')
     
     # activate beam search and early_stopping
     preds = model.generate(
@@ -53,6 +56,7 @@ def generate_preds(gen_mode, desc):
              }
     return pd.DataFrame(df)
 
+
 demo = gr.Blocks()
 with demo:
     gr.Markdown("<h1><center>RecipeGen: Automated TAPs Generation Tool</center></h1>")
@@ -76,7 +80,7 @@ with demo:
                 with gr.Row():
                     generate = gr.Button("Generate")
                 gr.Markdown("<h1><center>Results</center></h1>")
-                results = gr.Dataframe(headers=["Trigger", "Trigger Description", "Action", "Action Description"], row_count=100)
+                results = gr.Dataframe(headers=["Trigger", "Trigger Description", "Action", "Action Description"])
         
         with gr.TabItem("Field"):    
             with gr.Column():
@@ -88,5 +92,5 @@ with demo:
                 gr.Markdown("<h1><center>Results</center></h1>")
                 results_field = gr.Dataframe(headers=["Trigger", "Trigger Description", "Trigger Fields", "Action", "Action Description", "Action Fields"])
     
-    generate.click(generate_preds, inputs=[gen_mode, desc], outputs=[results])
+    generate.click(generate_taps, inputs=[gen_mode, desc], outputs=[results])
 demo.launch()