Removed TempScale class for debugging

app.py

@@ -1,22 +1,23 @@
 import gradio as gr
 import numpy as np
-from transformers import pipeline
+from transformers import pipeline, Pipeline
 
 unmasker = pipeline("fill-mask", model="anferico/bert-for-patents")
 
 example = 'A crustless sandwich made from two slices of baked bread'
 
 
-class TempScalePipe(pipeline):
-    def _forward(self, model_inputs):
-        outputs = self.model(**model_inputs)
-        return outputs
-
-    def postprocess(self, model_outputs, temp=1e3):
-        out = model_outputs["logits"] / temp
-        idx = np.random.randint(-3, 0)
-        best_class = out.softmax(idx)
-        return best_class
+# class TempScalePipe(Pipeline):
+#     def _forward(self, model_inputs):
+#         outputs = self.model(**model_inputs)
+#         return outputs
+#
+#     def postprocess(self, model_outputs, temp=1e3):
+#         out = model_outputs["logits"] / temp
+#         idx = np.random.randint(-3, 0)
+#         best_class = out.softmax(idx)
+#         return best_class
+#
 
 def unmask(text):
     text = add_mask(text)
@@ -27,7 +28,7 @@ def unmask(text):
 
 textbox = gr.Textbox(label="Type language here", lines=5)
 import gradio as gr
-from transformers import pipeline
+from transformers import pipeline, Pipeline
 
 
 unmasker = pipeline("fill-mask", model="anferico/bert-for-patents")