bad commited on
Commit
3e429ea
1 Parent(s): 0d42ab0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -1,9 +1,8 @@
1
  import cv2
2
  import numpy as np
3
- import os
4
  import paddle
5
- import paddle.optimizer
6
- from Generater import Generater,VGG19
7
 
8
 
9
  generator = Generater()
@@ -12,8 +11,6 @@ generator = Generater()
12
  G_path ='Gmodel_state33003.pdparams'
13
  layer_state_dictg = paddle.load(G_path)
14
  generator.set_state_dict(layer_state_dictg)#导入训练好的参数文件
15
- import gradio as gr
16
- import cv2
17
 
18
  def style_transfer(content_img,style_img):
19
  g_input = content_img.astype('float32') / 127.5 - 1 # 归一化
 
1
  import cv2
2
  import numpy as np
 
3
  import paddle
4
+ from Generater import Generater
5
+ import gradio as gr
6
 
7
 
8
  generator = Generater()
 
11
  G_path ='Gmodel_state33003.pdparams'
12
  layer_state_dictg = paddle.load(G_path)
13
  generator.set_state_dict(layer_state_dictg)#导入训练好的参数文件
 
 
14
 
15
  def style_transfer(content_img,style_img):
16
  g_input = content_img.astype('float32') / 127.5 - 1 # 归一化