Update app.py

app.py

@@ -6,31 +6,42 @@ from PIL import Image
 REPO_ID = "owaiskha9654/Yolov7_Custom_Object_Detection"
 FILENAME = "best.pt"
 
-yolov5_weights = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
+yolov7_custom_weights = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
 
-model = torch.hub.load('WongKinYiu/yolov7:main',model='custom', path_or_model=yolov5_weights, force_reload=True)  # local repo
+model = torch.hub.load('WongKinYiu/yolov7:main',model='custom', path_or_model=yolov7_custom_weights, force_reload=True)  # local repo
 
-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
+def object_detection(im, size=416):
+    results = model(im) 
+    results.render()  
     return Image.fromarray(results.imgs[0])
 
 title = "Yolov7 Custom"
-description = """Esse modelo é uma pequena demonstração baseada em uma análise de cerca de 60 imagens somente. Para resultados mais confiáveis e genéricos, são necessários mais exemplos (imagens).
-"""
 
-image = gr.inputs.Image(shape=(640, 640), image_mode="RGB", source="upload", label="Imagem", optional=False)
+image = gr.inputs.Image(shape=(416, 416), image_mode="RGB", source="upload", label="Upload Image", optional=False)
 outputs = gr.outputs.Image(type="pil", label="Output Image")
 
+Custom_description="<center>Custom Training Performed on Kaggle <a href='https://www.kaggle.com/code/owaiskhan9654/training-yolov7-on-kaggle-on-custom-dataset/notebook' style='text-decoration: underline' target='_blank'>Link</a> </center><br> <center>Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors </center> <br> <b>1st</b> class is for Person Detected<br><b>2nd</b> class is for Car Detected"
+
+Footer = (
+    "<center>Model Trained by: Owais Ahmad Data Scientist at <b> Thoucentric </b> <a href=\"https://www.linkedin.com/in/owaiskhan9654/\">Visit Profile</a> <br></center>"
+    
+    "<center> Model Trained Kaggle Kernel <a href=\"https://www.kaggle.com/code/owaiskhan9654/training-yolov7-on-kaggle-on-custom-dataset/notebook\">Link</a> <br></center>"
+        
+    "<center> Kaggle Profile <a href=\"https://www.kaggle.com/owaiskhan9654\">Link</a> <br> </center>"
+    
+    "<center> HuggingFace🤗 Model Deployed Repository <a href=\"https://huggingface.co/owaiskha9654/Yolov7_Custom_Object_Detection\">Link</a> <br></center>"
+)
+
+examples1=[["Image1.jpeg"],["Image2.jpeg"],["Image3.jpeg"],["Image4.jpeg"],["Image5.jpeg"],["Image6.jpeg"],["horses.jpeg"],["horses.jpeg"]]
+
+Top_Title="<center>Yolov7 🚀 Custom Trained by <a href='https://www.linkedin.com/in/owaiskhan9654/' style='text-decoration: underline' target='_blank'>Owais Ahmad </center></a>🚗Car and 👦Person Detection"
+
+
 gr.Interface(
     fn=object_detection,
     inputs=image,
     outputs=outputs,
-    title=title,
-    description=description,
-    examples=[["car-person-2.jpg"], ["car-person-2.jpg"]]
-              ,cache_examples=False
-).launch()
+    title=Top_Title,
+    description=Custom_description,
+    article=Footer,
+    examples=[["car-person-2.jpg"], ["car-person-2.jpg"]]).launch()