fix

app.py

@@ -4,6 +4,7 @@ import pandas as pd
 import streamlit as st
 import torch
 from torch import Tensor
+import torch.nn.functional as F
 from transformers import AutoModel, AutoTokenizer
 
 import os