Hugging Face's logo

Spaces:
Neurazum
/
Vbai-DPA-2.2c
Sleeping

App Files Community
eyupipler commited on
Commit
3c5d9ef
·
verified ·
1 Parent(s): 0298670

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +2 -0
model.py CHANGED
@@ -1,5 +1,7 @@
1
  import torch
2
  import torch.nn as nn
 
 
3
 
4
  class SimpleCNN(nn.Module):
5
  def __init__(self, num_classes=6):
 
1
  import torch
2
  import torch.nn as nn
3
+ from huggingface_hub import hf_hub_download
4
+
5
 
6
  class SimpleCNN(nn.Module):
7
  def __init__(self, num_classes=6):