Update main.py

main.py

@@ -1,3 +1,6 @@
+# https://medium.com/@qacheampong/building-and-deploying-a-fastapi-app-with-hugging-face-9210e9b4a713
+# https://huggingface.co/spaces/Queensly/FastAPI_in_Docker
+
 from fastapi import FastAPI
 import pickle
 import uvicorn
@@ -5,59 +8,33 @@ import pandas as pd
 
 app = FastAPI()
 
-# @app.get("/")
-# def read_root():
-#     return {"Hello": "World!"}
-
-
-# Function to load pickle file
-def load_pickle(filename):
-    with open(filename, 'rb') as file:
-        data = pickle.load(file)
-        return data
-    
-# Load pickle file
-ml_components = load_pickle('ml_sepsis.pkl') 
-
-# Components in the pickle file
-ml_model = ml_components['model']
-pipeline_processing = ml_components['pipeline']    
-
+ 
 #Endpoints 
 #Root endpoints
 @app.get("/")
 def root():
-    return {"API": "An API for Sepsis Prediction."}
+    return {"API": "Sum of 2 Squares"}
     
 @app.get('/Predict_Sepsis')
-async def predict(Plasma_glucose: int, Blood_Work_Result_1: int,
-                  Blood_Pressure: int, Blood_Work_Result_2: int,
-                    Blood_Work_Result_3: int, Body_mass_index: float, 
-                    Blood_Work_Result_4: float,Age: int, Insurance:float):
+async def predict(number_1: int, 
+                  number_2: int,:
     
-    data = pd.DataFrame({'Plasma glucose': [Plasma_glucose], 'Blood Work Result-1':	[Blood_Work_Result_1],
-                         'Blood Pressure': [Blood_Pressure], 'Blood Work Result-2': [Blood_Work_Result_2],
-                        'Blood Work Result-3': [Blood_Work_Result_3], 'Body mass index': [Body_mass_index],
-                        'Blood Work Result-4':	[Blood_Work_Result_4], 'Age': [Age], 'Insurance':[Insurance]})
     
-    data_prepared = pipeline_processing.transform(data)
-
-    model_output = ml_model.predict(data_prepared).tolist()
 
-    prediction = make_prediction(model_output)
+    prediction = number_1**2+number_2**2
 
     return prediction
     
     
 
 
-def make_prediction(data_prepared):
+# def make_prediction(data_prepared):
 
-    output_pred = data_prepared
+#     output_pred = data_prepared
 
-    if output_pred == 0:
-        output_pred = "Sepsis status is Negative"
-    else:
-        output_pred = "Sepsis status is Positive"
+#     if output_pred == 0:
+#         output_pred = "Sepsis status is Negative"
+#     else:
+#         output_pred = "Sepsis status is Positive"
         
-    return output_pred
+#     return output_pred