Horus7 commited on
Commit
6d69209
1 Parent(s): 23b3221

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -28
app.py CHANGED
@@ -12,9 +12,9 @@ def detect(img):
12
  prediction = np.around(model.predict(img)[0], decimals=0)[0]
13
 
14
  if prediction == 1:
15
- return "Pneumonia Not Detected!"
16
 
17
- return "Pneumonia Detected!"
18
 
19
 
20
  input = gr.inputs.Image(shape=(100, 100))
@@ -27,29 +27,3 @@ title = "PneumoDetect: Pneumonia Detection from Chest X-Rays"
27
 
28
  iface = gr.Interface(fn=detect, inputs=input, outputs="text", examples=examples, examples_per_page=20, title=title)
29
  iface.launch(inline=False)
30
-
31
-
32
- # model_name = "Horus7/kaduce" # Remplacez par le nom de votre modèle Hugging Face
33
-
34
- # # tokenizer = AutoTokenizer.from_pretrained(model_name)
35
- # model = AutoModelForSequenceClassification.from_pretrained(model_name)
36
- # def predict(image):
37
- # # Prétraiter l'image (vous pouvez adapter cela en fonction des besoins de votre modèle)
38
- # preprocessed_image = preprocess_image(image)
39
-
40
- # # Faire une prédiction avec le modèle
41
- # prediction = model.predict(preprocessed_image)
42
-
43
- # # Renvoyer la prédiction
44
- # return prediction
45
-
46
- # # Fonction pour prétraiter l'image avant la prédiction
47
- # def preprocess_image(image):
48
- # # Effectuer les étapes de prétraitement nécessaires pour votre modèle (redimensionnement, normalisation, etc.)
49
- # return image
50
-
51
- # # Interface Gradio
52
- # inputs = gr.inputs.Image() # Entrée : une image
53
- # outputs = gr.outputs.Label() # Sortie : une étiquette
54
-
55
- # gr.Interface(fn=predict, inputs=inputs, outputs=outputs).launch()
 
12
  prediction = np.around(model.predict(img)[0], decimals=0)[0]
13
 
14
  if prediction == 1:
15
+ return "Pneumonia Detected!"
16
 
17
+ return "Pneumonia Not Detected!"
18
 
19
 
20
  input = gr.inputs.Image(shape=(100, 100))
 
27
 
28
  iface = gr.Interface(fn=detect, inputs=input, outputs="text", examples=examples, examples_per_page=20, title=title)
29
  iface.launch(inline=False)