splendid commited on
Commit
836850f
·
1 Parent(s): 69e4ec4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -4,6 +4,8 @@ from translate import Translator
4
  import matplotlib.pyplot as plt
5
  import gradio as gr
6
  import tensorflow as tf
 
 
7
 
8
  model = keras_cv.models.StableDiffusion(img_width=512, img_height=512)
9
  translator = Translator(from_lang="ko", to_lang="en")
 
4
  import matplotlib.pyplot as plt
5
  import gradio as gr
6
  import tensorflow as tf
7
+ import os
8
+ os.environ['TF_ENABLE_ONEDNN_OPTS']='0'
9
 
10
  model = keras_cv.models.StableDiffusion(img_width=512, img_height=512)
11
  translator = Translator(from_lang="ko", to_lang="en")