valanga commited on
Commit
493443b
·
verified ·
1 Parent(s): 3d55d2f

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -53
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
-