Hugging Face's logo

Spaces:
indra-ipd
/
DR-classifier
Sleeping

App Files Community
10
indra-ipd commited on
Commit
d1ca099
·
1 Parent(s): 85cdc3d

update

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -31,8 +31,8 @@ def transform_image(img_sample):
31
  def predict(Image):
32
  tranformed_img = transform_image(Image)
33
  model.eval()
34
- tranformed_img = transform_image(Image)
35
- img = torch.from_numpy(tranformed_img)
36
 
37
  with torch.no_grad():
38
  grade = torch.softmax(model(img.float()), dim=1)[0]
 
31
  def predict(Image):
32
  tranformed_img = transform_image(Image)
33
  model.eval()
34
+ img = transform_image(Image)
35
+ #img = torch.from_numpy(tranformed_img)
36
 
37
  with torch.no_grad():
38
  grade = torch.softmax(model(img.float()), dim=1)[0]