Update def to use summarizer

.gitignore

@@ -0,0 +1,2 @@
+venv/
+venv

app.py

@@ -1,7 +1,19 @@
 import gradio as gr
+from transformers import pipeline
 
-def greet(name):
-    return "Hello " + name + "!!"
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+summarizer = pipeline('summarization', model='Eitanli/resume_label_summary_model')
+
+
+def predict(text):
+    summary = summarizer(text, max_length=10, min_length=2)
+    return {'summary': summary}
+
+
+iface = gr.Interface(
+    fn=predict,
+    inputs='text',
+    outputs="text")
+
+if __name__ == "__main__":
+    iface.launch()

requirements.txt

@@ -0,0 +1,2 @@
+transformers==4.30.2
+torch>=2.0