main.py

# 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
import pandas as pd

app = FastAPI()

 
#Endpoints 
#Root endpoints
@app.get("/")
def root():
    return {"API": "Sum of 2 Squares"}
    
@app.get('/Sum_Square')
async def predict(number_1: int, number_2: int):
    
    prediction = number_1**2 + number_2**2

    return prediction
    
    


# def make_prediction(data_prepared):

#     output_pred = data_prepared

#     if output_pred == 0:
#         output_pred = "Sepsis status is Negative"
#     else:
#         output_pred = "Sepsis status is Positive"
        
#     return output_pred