imagensAletorias

app.py

@@ -3,6 +3,7 @@ import numpy as np
 import keras
 from keras.preprocessing import image
 from huggingface_hub import from_pretrained_keras
+import random
 
 # Remove whitespace from the top of the page
 reduce_header_height_style = """
@@ -54,7 +55,11 @@ def inicializa_modelo():
 	model = from_pretrained_keras(file)		
 	print('modelo carregado.')
 	return model
-                     
+
+def escolher_imagem():
+	imagens = ['CHNCXR_0342_1.png', 'CHNCXR_0022_0.png', 'CHNCXR_0006_0.png', 'CHNCXR_0004_0.png', 'CHNCXR_0338_1.png', 'CHNCXR_0340_1.png', 'CHNCXR_0020_0.png', 'CHNCXR_0019_0.png', 'CHNCXR_0024_0.png', 'CHNCXR_0344_1.png', 'CHNCXR_0002_0.png', 'CHNCXR_0327_1.png', 'CHNCXR_0346_1.png', 'CHNCXR_0005_0.png', 'CHNCXR_0021_0.png', 'CHNCXR_0341_1.png', 'CHNCXR_0339_1.png', 'CHNCXR_0023_0.png', 'CHNCXR_0343_1.png', 'CHNCXR_0007_0.png', 'CHNCXR_0003_0.png', 'CHNCXR_0347_1.png', 'CHNCXR_0345_1.png', 'CHNCXR_0025_0.png', 'CHNCXR_0018_0.png', 'CHNCXR_0001_0.png', 'CHNCXR_0328_1.png', 'CHNCXR_0350_1.png', 'CHNCXR_0014_0.png', 'CHNCXR_0331_1.png', 'CHNCXR_0349_1.png', 'CHNCXR_0016_0.png', 'CHNCXR_0333_1.png', 'CHNCXR_0337_1.png', 'CHNCXR_0012_0.png', 'CHNCXR_0335_1.png', 'CHNCXR_0010_0.png', 'CHNCXR_0009_0.png', 'CHNCXR_0332_1.png', 'CHNCXR_0017_0.png', 'CHNCXR_0351_1.png', 'CHNCXR_0329_1.png', 'CHNCXR_0348_1.png', 'CHNCXR_0330_1.png', 'CHNCXR_0015_0.png', 'CHNCXR_0011_0.png', 'CHNCXR_0334_1.png', 'CHNCXR_0008_0.png', 'CHNCXR_0013_0.png', 'CHNCXR_0336_1.png']
+	imagem = random.sample(imagens, k=1)[0]	
+	return imagem
 
 if 'model' not in st.session_state:
 	with st.spinner('carregando o modelo...'):
@@ -72,6 +77,13 @@ with st.container():
 		name_image = "doencas_pulmonares.jpg"
 		imagem = image.load_img(name_image)
 		col1.image(imagem, width=700)
+		
+	if st.button('Raios-X Diversos'):
+		nome_imagem = escolher_imagem()
+		col1.image(imagem)
+		col1.header(predict_image(file_upload))
+
+
 
 st.write("")
 st.write("")