Update app.py

app.py

@@ -1,4 +1,8 @@
 import gradio as gr
+from fastai.vision.all import *
+import skimage 
+
+learn = load_learner('export.pkl')
 
 def greet(name):
     if name == 'Just happened a terrible car crash':
@@ -10,8 +14,10 @@ def greet(name):
          'input_ids': [1, 54453, 294, 1063, 1625, 266, 4838, 640, 4325, 2],
          'token_type_ids': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
          'attention_mask': [1, 1, 1, 1, 1, 1, 1, 1, 1, 1]}
+
+        pred,pred_idx,probs = learn.predict(map)
         
-        return "test " + name + "!"
+        return probs
     if name == 'Heard about #earthquake is different cities, stay safe everyone':
         map = {'id': 2,
          'keyword': '',
@@ -22,7 +28,9 @@ def greet(name):
          'token_type_ids': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
          'attention_mask': [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]}
 
-        return "Hello " + name + "!"
+        pred,pred_idx,probs = learn.predict(map)
+        
+        return probs
     if name == 'Theyd probably still show more life than Arsenal did yesterday, eh? EH?':
         map = {'id': 21,
          'keyword': '',
@@ -33,7 +41,9 @@ def greet(name):
          'token_type_ids': [0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0],
          'attention_mask': [1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1]}
 
-        return "Hello " + name + "!"
+        pred,pred_idx,probs = learn.predict(map)
+        
+        return probs
     return
 
 examples = [['Just happened a terrible car crash'],['Heard about #earthquake is different cities, stay safe everyone'], ['Theyd probably still show more life than Arsenal did yesterday, eh? EH?']]