Update app.py

app.py

@@ -39,7 +39,7 @@ def pass_through(torch_output, key: str):
         key: torch_output.type(torch.float32).to(device)
     }
     output = model(input_data)
-    return output[key]
+    return output[key].detach().numpy()
 
 
 def nt_embed(sequence: str):