Update app.py

app.py

@@ -1,4 +1,13 @@
 
+# Import the necessary libraries
+from transformers import AutoModel, AutoTokenizer
+import torch
+from torch.utils.data import DataLoader, Dataset
+from sklearn.model_selection import train_test_split
+from google.colab import files
+import pandas as pd
+import gradio as gr
+
 # Load the pre-trained model and tokenizer
 model = AutoModel.from_pretrained("Alibaba-NLP/gte-multilingual-base", trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained("Alibaba-NLP/gte-multilingual-base", trust_remote_code=True)
@@ -32,3 +41,4 @@ iface = gr.Interface(
 
 # Launch the Gradio interface
 iface.launch()
+