Update app.py

app.py

@@ -55,7 +55,7 @@ def predict(string):
 
   # Transform the raw dataset using non-batched API (i.e apply transformation line by line)
   def apply_transform(x):
-    return transform(x[0]), x[1]
+    return text_transform(x[0]), x[1]
 
   dp = dp.map(apply_transform)
   dp = dp.batch(1)