Spaces:
Sleeping
Sleeping
Upload 3 files
Browse files- gradi.py +9 -2
- model_CNN.zip +3 -0
- requirements.txt +4 -0
gradi.py
CHANGED
|
@@ -4,13 +4,20 @@ Created on Wed Apr 17 21:57:52 2024
|
|
| 4 |
|
| 5 |
@author: admin
|
| 6 |
"""
|
| 7 |
-
|
| 8 |
import gradio as gr
|
| 9 |
import tensorflow as tf
|
| 10 |
import numpy as np
|
| 11 |
|
| 12 |
# 加载 TensorFlow 模型
|
| 13 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
loaded_model = tf.saved_model.load(model_path)
|
| 15 |
model_predict = loaded_model.signatures['serving_default']
|
| 16 |
# 假设我们有一个处理这些输入的函数
|
|
|
|
| 4 |
|
| 5 |
@author: admin
|
| 6 |
"""
|
| 7 |
+
import zipfile
|
| 8 |
import gradio as gr
|
| 9 |
import tensorflow as tf
|
| 10 |
import numpy as np
|
| 11 |
|
| 12 |
# 加载 TensorFlow 模型
|
| 13 |
+
|
| 14 |
+
zip_path = 'model_CNN.zip' # 更改为您的压缩文件路径
|
| 15 |
+
extract_path = '/model' # 更改为您想要解压到的路径
|
| 16 |
+
|
| 17 |
+
# 解压缩文件
|
| 18 |
+
with zipfile.ZipFile(zip_path, 'r') as zip_ref:
|
| 19 |
+
zip_ref.extractall(extract_path)
|
| 20 |
+
model_path = extract_path + '/model_CNN'
|
| 21 |
loaded_model = tf.saved_model.load(model_path)
|
| 22 |
model_predict = loaded_model.signatures['serving_default']
|
| 23 |
# 假设我们有一个处理这些输入的函数
|
model_CNN.zip
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bd2f9fd8eb6cf548d46b65a31cbf02566b572e7026c00c096ad89033fdab0883
|
| 3 |
+
size 1048577
|
requirements.txt
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
Python version: 3.8.5
|
| 2 |
+
Gradio version: 4.26.0
|
| 3 |
+
TensorFlow version: 2.4.0
|
| 4 |
+
NumPy version: 1.19.5
|