Update app.py

app.py

@@ -1,3 +1,21 @@
 import gradio as gr
+import torch
+from transformers import pipeline
+
+#pipeline = pipeline(task="text-generation", model="EmailSubjectLineGeneration")
+pipeline = pipeline(model="jeevana/EmailSubjectLineGeneration")
+
+def predict(input):
+    print("pipeline object", pipeline)
+    prediction = pipeline(input)
+    print("type of response:", type(input))
+    return prediction 
+
+
+app = gr.Interface(fn=predict, inputs=[gr.Textbox(label="Email", lines=12)],
+                    outputs=[gr.Textbox(label="Subject", lines=3)],
+                    title="EmailSubjectLineGeneration",
+                    description="EmailSubjectLineGeneration"
+                   )
+app.launch(share=True, debug=True)
 
-gr.load("models/jeevana/EmailSubjectLineGeneration").launch()