Upload folder using huggingface_hub

app.py

@@ -1,7 +1,7 @@
 import joblib
 import numpy as np
 import pandas as pd
-from fastapi import FastAPI
+from fastapi import FastAPI, Request, UploadFile, File
 
 superkart_api = FastAPI(title="Superkart Sales Prediction",
               description="API for predicting Superkart sales",
@@ -14,20 +14,20 @@ def home():
   return "Welcome to SuperKart Sales Prediction API!"
 
 @superkart_api.post('/v1/superkart_single')
-def salepred_single():
-  sales_data = request.get_json()
+async def salepred_single(request: Request):
+  sales_data = await request.json()
 
   # Read input data
   sample = {
-      'Product_Weight':sale_data['Product_Weight'],
-      'Product_Sugar_Content':sale_data['Product_Sugar_Content'],
-      'Product_Allocated_Area':sale_data['Product_Allocated_Area'],
-      'Product_Type':sale_data['Product_Type'],
-      'Product_MRP':sale_data['Product_MRP'],
-      'Store_Id':sale_data['Store_Id'],
-      'Store_Size':sale_data['Store_Size'],
-      'Store_Location_City_Type':sale_data['Store_Location_City_Type'],
-      'Store_Type':sale_data['Store_Type'],
+      'Product_Weight':sales_data['Product_Weight'],
+      'Product_Sugar_Content':sales_data['Product_Sugar_Content'],
+      'Product_Allocated_Area':sales_data['Product_Allocated_Area'],
+      'Product_Type':sales_data['Product_Type'],
+      'Product_MRP':sales_data['Product_MRP'],
+      'Store_Id':sales_data['Store_Id'],
+      'Store_Size':sales_data['Store_Size'],
+      'Store_Location_City_Type':sales_data['Store_Location_City_Type'],
+      'Store_Type':sales_data['Store_Type'],
 
   }
   input_data = pd.DataFrame([sample])
@@ -37,15 +37,13 @@ def salepred_single():
 
   # Create response
   response = {'Store_Outlet':sample['Store_Id'],"Sale":round(float(predicted_sale), 2)}
-  return jsonify(response)
+  return response
 
 @superkart_api.post('/v1/superkart_batch')
-def salepred_batch():
-    file = request.files['file']
-    print("File Received:", file.filename)
+async def salepred_batch(file: UploadFile = File(...)):
 
     # Read input data
-    input_data = pd.read_csv(file)
+    input_data = pd.read_csv(file.file)
 
     # Make predictions
     predicted_sale = model.predict(input_data).tolist()
@@ -62,10 +60,4 @@ def salepred_batch():
     }
     print("Final Response:", response)
 
-    return jsonify(response)
-
-
-
-
-if __name__=='__main__':
-  superkart_api.run()
+    return response