Upload folder using huggingface_hub

app.py

@@ -50,11 +50,12 @@ def predict_batch():
 
     # Read the file into a DataFrame
     input_data = pd.read_csv(file)
+    id_list = input_data.Product_Id.values.tolist()
+    # Transform the input using the same trained pipeline:
     input_data = pipeline.transform(input_data)
     # Make predictions for the batch data:
     predictions = model.predict(input_data).tolist()
-
-    id_list = input_data.Product_Id.values.tolist()
+    
     output_dict = dict(zip(id_list, predictions))
 
     return output_dict