Delete app.py
Browse files
app.py
DELETED
@@ -1,53 +0,0 @@
|
|
1 |
-
from PIL import Image
|
2 |
-
import numpy as np
|
3 |
-
import gradio as gr
|
4 |
-
from tensorflow.keras.models import load_model
|
5 |
-
|
6 |
-
# Carica il modello salvato in formato Keras .h5
|
7 |
-
model = load_model('mio_modello.h5')
|
8 |
-
|
9 |
-
# Preprocessare l'immagine
|
10 |
-
def preprocess_image(image):
|
11 |
-
#image = Image.fromarray(image) # Assumendo che 'image' sia un array numpy
|
12 |
-
image = image.resize((64, 64)) # Ridimensionare l'immagine a 64x64
|
13 |
-
|
14 |
-
# Aggiungi un controllo per verificare la forma dell'immagine
|
15 |
-
print(f"Image shape after resizing and normalization: {image_array.shape}")
|
16 |
-
|
17 |
-
#image_array = np.expand_dims(image_array, axis=0) # Aggiungi una dimensione batch
|
18 |
-
return image
|
19 |
-
|
20 |
-
# Classificazione: Definire una funzione per fare la predizione con le percentuali di confidenza
|
21 |
-
def classify_image(image):
|
22 |
-
image_array = preprocess_image(image)
|
23 |
-
|
24 |
-
# Fare la predizione e prendere la prima riga dei risultati
|
25 |
-
prediction = model.predict(image_array)[0]
|
26 |
-
|
27 |
-
# Aggiungi un print per vedere le predizioni grezze
|
28 |
-
print(f"Raw model predictions: {prediction}")
|
29 |
-
|
30 |
-
predicted_class_idx = np.argmax(prediction) # Indice della classe con il punteggio più alto
|
31 |
-
|
32 |
-
# Classi (Chihuahua o Muffin)
|
33 |
-
class_labels = ['Chihuahua', 'Muffin']
|
34 |
-
|
35 |
-
# Creare un dizionario con le percentuali di confidenza
|
36 |
-
confidence_scores = {class_labels[i]: float(prediction[i]) for i in range(len(class_labels))}
|
37 |
-
|
38 |
-
return confidence_scores # Restituire le percentuali per ciascuna classe
|
39 |
-
|
40 |
-
# Definire l'interfaccia di Gradio
|
41 |
-
gr.Interface(
|
42 |
-
fn=classify_image, # Funzione di classificazione
|
43 |
-
inputs=gr.Image(),
|
44 |
-
outputs=gr.Label(num_top_classes=2), # Output: label con le percentuali per le due classi
|
45 |
-
title="Chihuahua vs Muffin Classifier",
|
46 |
-
description="Carica un'immagine e scopri se è un Chihuahua o un Muffin, con percentuali di confidenza!"
|
47 |
-
).launch()
|
48 |
-
|
49 |
-
|
50 |
-
|
51 |
-
|
52 |
-
|
53 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|