simayhosmeyve commited on
Commit
985c33c
1 Parent(s): d29b6ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -428,8 +428,7 @@ plt.ylabel('Loss Percent')
428
  plt.xlabel('Epochs')
429
  plt.show()
430
 
431
- from transformers import BertConfig, TFBertModel
432
- pre_trained = TFBertModel.from_pretrained("gradio_pix2pix.h5")
433
  #pre_trained = keras.models.load("gradio_pix2pix.h5")
434
 
435
  #pre_trained.summary()
 
428
  plt.xlabel('Epochs')
429
  plt.show()
430
 
431
+ pre_trained = Generator.from_pretrained("gradio_pix2pix.h5")
 
432
  #pre_trained = keras.models.load("gradio_pix2pix.h5")
433
 
434
  #pre_trained.summary()