Update app.py

app.py

@@ -21,15 +21,16 @@ class DotProductBias(Module):
         return sigmoid_range(res, *self.y_range)
 
 def get_y(r):
-    number = ''.join(filter(str.isdigit, str(r)))
-    return train_y[int(number)]
-
-def persistent_load(persid):
-    return None
-
-
-with open('collaborativefiltering.pkl', 'rb') as f:
-    learn = pickle.load(f)
+    result = learn.forward(r)
+    return result[0][0]
+
+class CustomUnpickler(pickle.Unpickler):
+    def persistent_load(self, persid):
+        return None
+    
+with open('CollaborativeFiltering.pkl', 'rb') as f:
+    custom_unpickler = CustomUnpickler(f)
+    learn = custom_unpickler.load()
 
 label = gr.outputs.Label()