wjc23 commited on
Commit
fac070e
·
verified ·
1 Parent(s): 9d97e04

Upload 3 files

Browse files
Files changed (3) hide show
  1. gradi.py +9 -2
  2. model_CNN.zip +3 -0
  3. 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
- model_path = 'model/model_CNN'
 
 
 
 
 
 
 
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