Update app.py

app.py

@@ -1,4 +1,16 @@
 import os
+import subprocess
+
+# Function to install a package if it is not already installed
+def install(package):
+    subprocess.check_call([os.sys.executable, "-m", "pip", "install", package])
+
+# Ensure the necessary packages are installed
+install("transformers")
+install("torch")
+install("pandas")
+install("gradio")
+
 import pandas as pd
 import gradio as gr
 from transformers import AutoModel, AutoTokenizer
@@ -26,8 +38,7 @@ def process_with_model(pec_number):
     inputs = tokenizer(pec_number, return_tensors="pt")
     with torch.no_grad():
         outputs = model(**inputs)
-    # Here, we simply return the last hidden state as a string representation
-    # In a real application, you might want to use this in a more meaningful way
+    # Return a simple representation of the model's output
     return outputs.last_hidden_state.mean(dim=1).squeeze().tolist()
 
 # Combine both functions to create a prediction
@@ -52,4 +63,3 @@ iface = gr.Interface(
 if __name__ == "__main__":
     iface.launch()
 
-