aznasut commited on
Commit
9cf84ce
·
1 Parent(s): 33dfaa8

ignore temp files

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -99,7 +99,7 @@ async def classify_image(file: UploadFile = File(None)):
99
 
100
  image = Image.open(io.BytesIO(image_data))
101
 
102
- inputs = image_processor(image, return_tensors="pt")
103
 
104
  with torch.no_grad():
105
  logits = model(**inputs).logits
@@ -184,7 +184,7 @@ async def classify_images(request: ImageUrlsRequest):
184
  continue
185
 
186
  image = Image.open(io.BytesIO(image_data))
187
- inputs = image_processor(image, return_tensors="pt")
188
 
189
  with torch.no_grad():
190
  logits = model(**inputs).logits
 
99
 
100
  image = Image.open(io.BytesIO(image_data))
101
 
102
+ inputs = model(image)
103
 
104
  with torch.no_grad():
105
  logits = model(**inputs).logits
 
184
  continue
185
 
186
  image = Image.open(io.BytesIO(image_data))
187
+ inputs = model(image)
188
 
189
  with torch.no_grad():
190
  logits = model(**inputs).logits