File size: 610 Bytes
7ca35c3
 
 
 
 
55ecf9a
7ca35c3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
from PIL import Image
import numpy as np
import gradio as gr
import keras

model = keras.models.load_model('CATvsDOG.keras')

def classify_image(image):
    img = Image.open(image)

    # Preprocess the image
    img = img.resize((100, 100))
    img_array = np.array(img)
    img_array = np.expand_dims(img_array, axis=0)

    # Make predictions
    predictions = model.predict(img_array)
    if predictions[0][0] > 0.5:
        return "Dog"
    else:
        return "Cat"

interface = gr.Interface(fn=classify_image, inputs="file", outputs="label", title="Cat or Dog Classifier")

interface.launch(share=True)