cyberai-1 commited on
Commit ·
a0faa4d
1
Parent(s): 548a2fb
Add
Browse files
app.py
CHANGED
|
@@ -88,14 +88,6 @@ def load_pytorch():
|
|
| 88 |
return _pytorch_model
|
| 89 |
|
| 90 |
|
| 91 |
-
def load_tensorflow():
|
| 92 |
-
global _tf_model
|
| 93 |
-
if _tf_model is None:
|
| 94 |
-
import tensorflow as tf
|
| 95 |
-
_tf_model = tf.keras.models.load_model("parfait_model.keras")
|
| 96 |
-
return _tf_model
|
| 97 |
-
|
| 98 |
-
|
| 99 |
def read_input_image():
|
| 100 |
if "image" in request.files and request.files["image"].filename:
|
| 101 |
return Image.open(io.BytesIO(request.files["image"].read())).convert("RGB")
|
|
|
|
| 88 |
return _pytorch_model
|
| 89 |
|
| 90 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 91 |
def read_input_image():
|
| 92 |
if "image" in request.files and request.files["image"].filename:
|
| 93 |
return Image.open(io.BytesIO(request.files["image"].read())).convert("RGB")
|