Update app.py

app.py

@@ -4,6 +4,7 @@ from transformers import AutoTokenizer, AutoModel
 # Use a pipeline as a high-level helper
 from transformers import pipeline
 import csv
+import torch
 
 pipe = pipeline("fill-mask", model="microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext")# Load the PubMedBERT tokenizer and model directly
 tokenizer = AutoTokenizer.from_pretrained("microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext")