GranularFireplace commited on
Commit
87891c6
·
verified ·
1 Parent(s): eb27cf6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -374,8 +374,10 @@ async def analyse_bin(file_name: str):
374
  convert_binary_to_image(file_path, temp_image, width)
375
  img_array = await process_image(temp_image)
376
  result = predict_malware(img_array)
377
- result['image'] = temp_image
378
- return result
 
 
379
 
380
  except HTTPException as he:
381
  raise he
 
374
  convert_binary_to_image(file_path, temp_image, width)
375
  img_array = await process_image(temp_image)
376
  result = predict_malware(img_array)
377
+ return {
378
+ "result": result,
379
+ "image_location": str(temp_image)
380
+ }
381
 
382
  except HTTPException as he:
383
  raise he