HawkeyeHS commited on
Commit
1310eff
·
1 Parent(s): 2c1e5b5

Add application file

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -39,7 +39,6 @@ def predict():
39
  try:
40
  image = Image.open(BytesIO(response.content))
41
  image = image.resize((128, 128))
42
- image.save("new.jpg")
43
  encoding = feature_extractor(image.convert("RGB"), return_tensors="pt")
44
  with torch.no_grad():
45
  outputs = model(**encoding)
 
39
  try:
40
  image = Image.open(BytesIO(response.content))
41
  image = image.resize((128, 128))
 
42
  encoding = feature_extractor(image.convert("RGB"), return_tensors="pt")
43
  with torch.no_grad():
44
  outputs = model(**encoding)