Update app.py

app.py

@@ -8,15 +8,12 @@ from haystack import Document
 from haystack.components.readers import ExtractiveReader
 import wikipediaapi
 
-def init_models():
-  model = ViTForImageClassification.from_pretrained("Dmitry43243242/banana-disease-leaf-model")
-  feature_extractor = ViTFeatureExtractor.from_pretrained("Dmitry43243242/banana-disease-leaf-model")
-  tokenizer_qa = AutoTokenizer.from_pretrained("deepset/roberta-base-squad2")
-  model_qa = AutoModelForQuestionAnswering.from_pretrained("deepset/roberta-base-squad2")
-  summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
-  translator = pipeline("translation", model="facebook/nllb-200-distilled-600M", torch_dtype=torch.bfloat16)
-
-  return model, feature_extractor, tokenizer_qa, model_qa, summarizer, translator
+model = ViTForImageClassification.from_pretrained("Dmitry43243242/banana-disease-leaf-model")
+feature_extractor = ViTFeatureExtractor.from_pretrained("Dmitry43243242/banana-disease-leaf-model")
+tokenizer_qa = AutoTokenizer.from_pretrained("deepset/roberta-base-squad2")
+model_qa = AutoModelForQuestionAnswering.from_pretrained("deepset/roberta-base-squad2")
+summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+translator = pipeline("translation", model="facebook/nllb-200-distilled-600M", torch_dtype=torch.bfloat16)
 
 def translate_question(question, translator):
   text_translated = translator(question,
@@ -42,8 +39,6 @@ def process_input(img, text):
   title="Banana Leaf Disease Classifier"
   labels = ['banana_healthy_leaf', 'black_sigatoka', 'yellow_sigatoka', 'panama_disease', 'moko_disease', 'insect_pest', 'bract_mosaic_virus']
 
-  model, feature_extractor, tokenizer_qa, model_qa, summarizer, translator = init_models()
-
 
   translate_eng_question = translate_question(text, translator)