nikolis commited on
Commit
ce04927
·
1 Parent(s): 1e8865c
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from huggingface_hub import InferenceClient
4
  app = FastAPI()
5
 
6
  MODEL_NAME = "flax-community/t5-recipe-generation"
7
- client = InferenceClient(model=MODEL_NAME, token=API_KEY)
8
 
9
  @app.post("/generate-recipe/")
10
  def generate_recipe(ingredients: str):
 
4
  app = FastAPI()
5
 
6
  MODEL_NAME = "flax-community/t5-recipe-generation"
7
+ client = InferenceClient(model=MODEL_NAME, token=os.getenv('API_KEY'))
8
 
9
  @app.post("/generate-recipe/")
10
  def generate_recipe(ingredients: str):