Update app.py

app.py

@@ -21,8 +21,8 @@ tokenizer = AutoTokenizer.from_pretrained("Alibaba-NLP/gte-multilingual-base", t
 model = AutoModel.from_pretrained("Alibaba-NLP/gte-multilingual-base", trust_remote_code=True)
 
 # Load the dataset containing PEC numbers and names
-def load_dataset(file_path=None):
-    if file_path and os.path.exists(file_path):
+def load_dataset(file_path='PEC_Numbers_and_Names.xlsx'):
+    if os.path.exists(file_path):
         df = pd.read_excel(file_path)
     else:
         raise FileNotFoundError(f"File not found: {file_path}")
@@ -31,7 +31,7 @@ def load_dataset(file_path=None):
 # Function to get the name based on the PEC number
 def get_name(pec_number, df):
     df['PEC No.'] = df['PEC No.'].str.strip().str.upper()
-    pec_number = pec_number.strip().upper()
+    pec_number = pec_number.strip().str.upper()
     
     result = df[df['PEC No.'] == pec_number]
     
@@ -48,13 +48,10 @@ def process_with_model(pec_number):
     return outputs.last_hidden_state.mean(dim=1).squeeze().tolist()
 
 # Combine both functions to create a prediction
-def predict(pec_number, file):
+def predict(pec_number):
     try:
-        # Load the dataset from the uploaded file if provided
-        if file is not None:
-            df = pd.read_excel(file.name)
-        else:
-            return "Please upload the PEC Numbers and Names file."
+        # Load the dataset from the root directory
+        df = load_dataset()
 
         name = get_name(pec_number, df)
         model_output = process_with_model(pec_number)
@@ -62,16 +59,13 @@ def predict(pec_number, file):
     except FileNotFoundError as e:
         return str(e)
 
-# Build the Gradio interface with file upload option
+# Build the Gradio interface without the file upload option
 iface = gr.Interface(
     fn=predict,
-    inputs=[
-        gr.Textbox(lines=1, placeholder="Enter PEC Number..."),
-        gr.File(label="Upload PEC Numbers and Names file")
-    ],
+    inputs=gr.Textbox(lines=1, placeholder="Enter PEC Number..."),
     outputs="text",
     title="PEC Number to Name Lookup",
-    description="Enter a PEC number to retrieve the corresponding name and process it with a Hugging Face model. Please upload the Excel file."
+    description="Enter a PEC number to retrieve the corresponding name and process it with a Hugging Face model."
 )
 
 # Run the Gradio interface