Spaces:

raihanp
/

Garbage-Type-Classifier

Sleeping

raihanp commited on Sep 19, 2024

Commit

f5de5fe

verified ·

1 Parent(s): 7b609a5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,6 +1,8 @@
 from PIL import Image
 import torch
 import gradio as gr
 inference = torch.load('fine_tune_resnet.pth')
 inference.eval()

 from PIL import Image
 import torch
 import gradio as gr
+from torchvision.transforms import Compose, Normalize, ToTensor, Resize, CenterCrop
 inference = torch.load('fine_tune_resnet.pth')
 inference.eval()