First commit

app.py

@@ -0,0 +1,40 @@
+import torchvision
+import torch
+from torch import nn
+from PIL import Image
+from torchvision import transforms
+import onnxruntime
+import numpy as np
+import torch.nn.functional as F
+from safetensors.torch import save_file,load_file,safe_open
+import numpy as np
+
+def predict(img_path,model=None):
+    if model is None:
+        pretrained_weights_resnet18=torchvision.models.ResNet18_Weights.DEFAULT
+        model=torchvision.models.resnet18(weights=pretrained_weights_resnet18)
+        class_names=pretrained_weights_resnet18.meta["categories"]
+    transform=transforms.Compose([transforms.Resize((64,64)),transforms.ToTensor(),transforms.Normalize(mean=[0.485,0.456,0.406],std=[0.229,0.224,0.225])])
+    if isinstance(img_path,np.ndarray):
+        image=Image.fromarray(img_path).convert("RGB")
+    else:
+        image=Image.open(img_path).convert("RGB")
+    img_transform=transform(image).unsqueeze(0)
+
+    model.eval()
+    with torch.inference_mode():
+        logit=model(img_transform)
+        pred_prob=torch.softmax(logit,dim=1).squeeze().numpy()
+        predict_dict={}
+        for i in range(len(class_names)):
+            predict_dict[class_names[i]]=float(pred_prob[i])
+        
+    return predict_dict
+
+
+import numpy as np
+import gradio as gr
+
+demo = gr.Interface(predict, gr.Image(), outputs=gr.Label(num_top_classes=3))
+if __name__ == "__main__":
+    demo.launch()

requirments.txt

@@ -0,0 +1,108 @@
+aiofiles==24.1.0
+annotated-types==0.7.0
+anyio==4.9.0
+asttokens==3.0.0
+certifi==2025.1.31
+charset-normalizer==3.4.1
+click==8.1.8
+colorama==0.4.6
+coloredlogs==15.0.1
+comm==0.2.2
+contourpy==1.3.2
+cycler==0.12.1
+debugpy==1.8.14
+decorator==5.2.1
+dnspython==2.7.0
+email_validator==2.2.0
+executing==2.2.0
+fastapi==0.115.12
+fastapi-cli==0.0.7
+ffmpy==0.5.0
+filelock==3.18.0
+flatbuffers==25.2.10
+fonttools==4.57.0
+fsspec==2025.3.2
+gradio==5.25.2
+gradio_client==1.8.0
+groovy==0.1.2
+h11==0.14.0
+httpcore==1.0.7
+httptools==0.6.4
+httpx==0.28.1
+huggingface-hub==0.30.2
+humanfriendly==10.0
+idna==3.10
+ipykernel==6.29.5
+ipython==9.1.0
+ipython_pygments_lexers==1.1.1
+jedi==0.19.2
+Jinja2==3.1.6
+jupyter_client==8.6.3
+jupyter_core==5.7.2
+kiwisolver==1.4.8
+mangum==0.19.0
+markdown-it-py==3.0.0
+MarkupSafe==3.0.2
+matplotlib==3.10.1
+matplotlib-inline==0.1.7
+mdurl==0.1.2
+mpmath==1.3.0
+nest-asyncio==1.6.0
+networkx==3.4.2
+numpy==2.2.4
+onnxruntime==1.21.1
+orjson==3.10.16
+packaging==24.2
+pandas==2.2.3
+parso==0.8.4
+pillow==11.2.1
+platformdirs==4.3.7
+prompt_toolkit==3.0.51
+protobuf==6.30.2
+psutil==7.0.0
+pure_eval==0.2.3
+pydantic==2.11.3
+pydantic_core==2.33.1
+pydub==0.25.1
+Pygments==2.19.1
+pyparsing==3.2.3
+pyreadline3==3.5.4
+python-dateutil==2.9.0.post0
+python-dotenv==1.1.0
+python-multipart==0.0.20
+pytz==2025.2
+pywin32==310
+PyYAML==6.0.2
+pyzmq==26.4.0
+regex==2024.11.6
+requests==2.32.3
+rich==14.0.0
+rich-toolkit==0.14.1
+ruff==0.11.6
+safehttpx==0.1.6
+safetensors==0.5.3
+semantic-version==2.10.0
+shellingham==1.5.4
+six==1.17.0
+sniffio==1.3.1
+stack-data==0.6.3
+starlette==0.46.1
+sympy==1.13.1
+tokenizers==0.21.1
+tomlkit==0.13.2
+torch==2.6.0
+torchinfo==1.8.0
+torchvision==0.21.0
+tornado==6.4.2
+tqdm==4.67.1
+traitlets==5.14.3
+transformers==4.51.3
+typer==0.15.2
+typing-inspection==0.4.0
+typing_extensions==4.13.2
+tzdata==2025.2
+urllib3==2.4.0
+uvicorn==0.34.0
+watchfiles==1.0.5
+wcwidth==0.2.13
+websockets==15.0.1