Update app.py

app.py

@@ -9,9 +9,7 @@ from resultados_consulta import resultados_consulta, detalles_resultados
 import tensorflow as tf
 import tensorflow.python.ops.numpy_ops.np_config as np_config
 from math import ceil
-import torch
 
-np_config.enable_numpy_behavior(prefer_float32=False, dtype_conversion_mode='all')
 
 def split_frame(input_df, rows):
     df=[]
@@ -140,8 +138,8 @@ def main():
                answer=tokenizer.decode(predict_answer_tokens) 
 
                if (len(answer)>0):
-                   answer_start_scores = torch.nn.functional.softmax(outputs.start_logits)
-                   answer_end_scores = torch.nn.functional.softmax(outputs.end_logits)
+                   answer_start_scores = tf.nn.softmax(outputs.start_logits)
+                   answer_end_scores = tf.nn.softmax(outputs.end_logits)
                    score = answer_start_scores*answer_end_scores
                    st.write(score)
                    cantidad_respuestas = cantidad_respuestas + 1