| import gradio as gr |
| from transformers import pipeline |
|
|
| examples = ["examples/example_0.jpg", |
| "examples/example_1.png", |
| "examples/example_2.png", |
| "examples/example_3.png", |
| "examples/example_4.jpg", |
| "examples/example_5.png"] |
|
|
| pipe = pipeline(task="image-classification", |
| model="Kaludi/csgo-weapon-classification") |
| gr.Interface.from_pipeline(pipe, |
| title="CSGO Weapon Image Classification", |
| description = "This is a CSGO Weapon Classifier Model that has been trained by <strong><a href='https://huggingface.co/Kaludi'>Kaludi</a></strong> to recognize <strong>11</strong> different types of Counter-Strike: Global Offensive (CSGO) Weapons, which include <strong>AK-47,AWP,Famas,Galil-AR,Glock,M4A1,M4A4,P-90,SG-553,UMP,USP</strong>. The model is capable of accurately classifying the weapon name present in an image. With its deep understanding of the characteristics of each weapon in the game, the model is a valuable tool for players and fans of CSGO.", |
| article = "<p style='text-align: center'><a href='https://github.com/Kaludii'>Github</a> | <a href='https://huggingface.co/Kaludi'>HuggingFace</a></p>", |
| examples=examples, |
| ).launch() |
