DHEIVER commited on
Commit
e6e90d9
·
1 Parent(s): f92fcff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -7
app.py CHANGED
@@ -1,17 +1,16 @@
1
  import os
 
 
 
 
2
 
3
  # Desabilitar o uso da GPU pelo TensorFlow
4
  os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
5
-
6
- # Resto do seu código TensorFlow aqui...
7
-
8
- import gradio as gr
9
- from keras.models import load_model
10
- from keras.preprocessing.image import load_img, img_to_array
11
 
12
  # Carregar o modelo treinado
13
  model_path = "50_epochs.h5"
14
- model = load_model(model_path)
15
 
16
  # Função para fazer a previsão usando o modelo
17
  def classify_image(image):
 
1
  import os
2
+ import numpy as np
3
+ import gradio as gr
4
+ import tensorflow as tf
5
+ from keras.preprocessing.image import load_img, img_to_array
6
 
7
  # Desabilitar o uso da GPU pelo TensorFlow
8
  os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
9
+ os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2" # Suprimir mensagens de aviso do TensorFlow
 
 
 
 
 
10
 
11
  # Carregar o modelo treinado
12
  model_path = "50_epochs.h5"
13
+ model = tf.keras.models.load_model(model_path)
14
 
15
  # Função para fazer a previsão usando o modelo
16
  def classify_image(image):