Upload folder using huggingface_hub

app.py

@@ -60,9 +60,8 @@ def predict_sales_batch():
     predicted_sales = [round(float(sales),2) for sales in model.predict(input_data).tolist()]
 
     # Create a dictionary of predictions with Product ID and Predicted sales
-    #product_id = input_data['Product_ID'].tolist()  # Assuming id as the key or product id
-    #output_dict = dict(zip(product_id, predicted_sales))  # Sales value
-    output_dict = dict(predicted_sales)  # Sales value
+    product_id = input_data['Product_ID'].tolist()  # Assuming id as the key or product id
+    output_dict = dict(zip(product_id, predicted_sales))  # Sales value
 
     # Return the predictions dictionary as a JSON response
     return output_dict