testing prediction

app.py

@@ -1,8 +1,15 @@
 import gradio as gr
+from transformers import pipeline
+
+pipeline = pipeline(task="question-answering",model="Intel/bert-base-uncased-squadv1.1-sparse-80-1x4-block-pruneofa")
 
 def greet(name):
     return "Hello " + name + "!!"
 
+def predict(text):
+    predictions = pipeline(text)
+    return predictions
+
 md = """
 App coming soon! 
 
@@ -11,10 +18,11 @@ Based on the [Prune Once for All: Sparse Pre-Trained Language Models](https://ar
 """
 
 iface = gr.Interface(
-    fn=greet, 
+    fn=predict, 
     inputs="text", 
     outputs="text",
     title = "Question & Answer with Sparse BERT using the SQuAD dataset",
     description = md
     )
+
 iface.launch()