Spaces:
Sleeping
Sleeping
from fastapi import FastAPI | |
from transformers import pipeline | |
app = FastAPI(docs_url="/") | |
def calculate_food_endpoint(activity: str, weight: int): | |
score = 0 | |
if activity == "Typical" : | |
score = 110 | |
elif activity == "Active" : | |
score = 125 | |
elif activity == "Overweight" : | |
score = 70 | |
elif activity == "Highly Active" : | |
score = 175 | |
elif activity == "Senior, neutered, inactive" : | |
score = 90 | |
elif activity == "Working Dog (light duty)" : | |
score = 130 | |
elif activity == "Working Dog (moderate duty)" : | |
score = 150 | |
elif activity == "Working Dog (heavy duty)" : | |
score = 175 | |
else : | |
score = 110 | |
return round( score / weight, 2) | |
""" | |
Calculates the recommended amount of dog food based on activity level and weight. | |
Args: | |
activity: The dog's activity level, as a number from 1 to 5. | |
weight: The dog's weight in kilograms. | |
Returns: | |
A JSON object containing the recommended amount of food in cups. | |
""" | |
# Check if the activity and weight parameters are present in the API request. | |
if activity is None or weight is None: | |
return {"error": "Please provide both activity level and weight."} | |
# Calculate the recommended amount of food. | |
result = calculate_food(activity, weight) | |
# Respond with the result as JSON. | |
return result |