Lollo9898 commited on
Commit
f32ae5c
·
verified ·
1 Parent(s): b86c1ba

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -72,7 +72,8 @@ transform = transforms.Compose([
72
  model.eval()
73
  model.model.to(device)
74
  y_pred= []
75
- img = Image.open('/work/horizon_ria_elsa/generated_data/fake-images-pos/part-000000/206496010652.png').convert('RGB')
 
76
 
77
  with torch.no_grad():
78
  # in_tens = model.processor(img, return_tensors='pt')['pixel_values']
 
72
  model.eval()
73
  model.model.to(device)
74
  y_pred= []
75
+ # Put your image to tes
76
+ img = Image.open('206496010652.png').convert('RGB')
77
 
78
  with torch.no_grad():
79
  # in_tens = model.processor(img, return_tensors='pt')['pixel_values']