NathanSegers commited on
Commit
8db4042
·
1 Parent(s): 0a643c5

Import change

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@ from fastapi import FastAPI, File, UploadFile
2
  from fastapi.middleware.cors import CORSMiddleware
3
  import numpy as np
4
  from PIL import Image
5
- from tensorflow import keras
6
  import os
7
  os.environ["KERAS_BACKEND"] = "tensorflow"
8
 
@@ -24,7 +24,7 @@ def greet_json():
24
  return {"Hello": "World!"}
25
 
26
 
27
- model = keras.saving.load_model("hf://nathansegers/masterclass-2025")
28
 
29
  @app.post('/upload/image')
30
  async def uploadImage(img: UploadFile = File(...)):
 
2
  from fastapi.middleware.cors import CORSMiddleware
3
  import numpy as np
4
  from PIL import Image
5
+ from tensorflow.keras.models import load_model
6
  import os
7
  os.environ["KERAS_BACKEND"] = "tensorflow"
8
 
 
24
  return {"Hello": "World!"}
25
 
26
 
27
+ model = load_model("hf://nathansegers/masterclass-2025")
28
 
29
  @app.post('/upload/image')
30
  async def uploadImage(img: UploadFile = File(...)):