Spaces:

Aravindan
/

butterfly_classification

Runtime error

Aravindan commited on Aug 6, 2022

Commit

7d5de97

1 Parent(s): 878c8b2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,6 +1,7 @@
 import cv2, torch
 import gradio as gr
 import numpy as np
 import torchvision.models as models
 from torchvision import transforms as T

 import cv2, torch
 import gradio as gr
 import numpy as np
+import torch.nn as nn
 import torchvision.models as models
 from torchvision import transforms as T