Update getReaction.py
Browse files- getReaction.py +1 -2
getReaction.py
CHANGED
|
@@ -11,7 +11,6 @@ model = Reaction(ckpt_path, device=torch.device('cpu'))
|
|
| 11 |
device = torch.device('cpu')
|
| 12 |
|
| 13 |
def get_reaction(image_path: str) -> list:
|
| 14 |
-
'''Returns a list of reactions extracted from the image.'''
|
| 15 |
image_file = image_path
|
| 16 |
return json.dumps(model.predict_image_file(image_file, molscribe=True, ocr=True))
|
| 17 |
|
|
@@ -65,7 +64,7 @@ def generate_combined_image(predictions, image_file):
|
|
| 65 |
ax.imshow(img)
|
| 66 |
ax.axis('off')
|
| 67 |
if idx < n_images:
|
| 68 |
-
ax.set_title(f"
|
| 69 |
|
| 70 |
# 删除多余的子图
|
| 71 |
for idx in range(n_images, len(axes)):
|
|
|
|
| 11 |
device = torch.device('cpu')
|
| 12 |
|
| 13 |
def get_reaction(image_path: str) -> list:
|
|
|
|
| 14 |
image_file = image_path
|
| 15 |
return json.dumps(model.predict_image_file(image_file, molscribe=True, ocr=True))
|
| 16 |
|
|
|
|
| 64 |
ax.imshow(img)
|
| 65 |
ax.axis('off')
|
| 66 |
if idx < n_images:
|
| 67 |
+
ax.set_title(f"### Reaction {idx + 1} ###",fontsize=42)
|
| 68 |
|
| 69 |
# 删除多余的子图
|
| 70 |
for idx in range(n_images, len(axes)):
|