import error corrected

model_functions.py

@@ -1,12 +1,13 @@
 import torch 
 from torch.nn import CrossEntropyLoss, MSELoss
 import re
+from models import disease_model
 #!pip install rouge_score
 
 #from rouge_score import rouge_scorer
 
 from nltk.translate.meteor_score import meteor_score
-from .models import disease_model
+
 
 def forward_batch(images, input_ids, attention_mask, answers, question_classes=None,qtype_classifier=None,fusion_module=None,q_types=None,q_types_mapping=None,task_heads=None,device=None,image_encoder=None,question_encoder=None):
     # Image encoding