Update app.py

app.py

@@ -1,37 +1,32 @@
-import torch
-import gradio as gr
-from huggingface_hub import hf_hub_download
-from PIL import Image
-pip install -qr https://raw.githubusercontent.com/ultralytics/yolov5/master/requirements.txt  #
+import torch as tf
 
-REPO_ID = "Kr1n3/Fashion-Items-Classification"
-FILENAME = "best.pt"
+import gradio as gr
 
-yolov5_weights = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
+inception_net = tf.keras.models.load_model('best.pt')  # load the model
 
-model = torch.hub.load('ultralytics/yolov5', 'custom', path=yolov5_weights, force_reload=True)  # local repo
+labels = ['bom', 'ruim']
 
-def object_detection(im, size=640):
-    results = model(im)  # inference
-    #results.print()  # print results to screen
-    #results.show()  # display results
-    #results.save()  # save as results1.jpg, results2.jpg... etc.
-    results.render()  # updates results.imgs with boxes and labels
-    return Image.fromarray(results.imgs[0])
 
 title = "Fashion Items Classification"
 description = """
 """
 
+def classify_image(inp):
+    inp = inp.reshape((-1, 640, 640, 3))
+    inp = tf.keras.applications.mobilenet_v2.preprocess_input(inp)
+    prediction = inception_net.predict(inp).flatten()
+    return (labels[1] if float(prediction) >= 0 else labels[0])
+
+
 image = gr.inputs.Image(shape=(640, 640), image_mode="RGB", source="upload", label="Imagem", optional=False)
-outputs = gr.outputs.Image(type="pil", label="Output Image")
+label = gr.outputs.Textbox(type="auto", label="Classificação")
 
 gr.Interface(
-    fn=object_detection,
+    fn=classify_image,
     inputs=image,
-    outputs=outputs,
+    outputs=label,
     title=title,
     description=description,
     examples=[["https://github.com/Kr1n3/MPC_2022/blob/main/dataset/pants_30.jpeg?raw=true"], ["https://github.com/Kr1n3/MPC_2022/blob/main/dataset/bag_01.jpg?raw=true"], 
-              ["https://github.com/Kr1n3/MPC_2022/blob/main/dataset/pants_33.jpg?raw=true"], ["https://github.com/Kr1n3/MPC_2022/blob/main/dataset/bag_14.JPG?raw=true"]],
-).launch()
+              ["https://github.com/Kr1n3/MPC_2022/blob/main/dataset/bag_14.JPG?raw=true"], ["https://github.com/Kr1n3/MPC_2022/blob/main/dataset/dress_45.JPG?raw=true"]],
+).launch()