Update app.py

app.py

@@ -83,8 +83,9 @@ def load_summarizer():
     return summarizer
 
 # generator, tokenizer = load_generator()
-generator = pipeline("text-generation", model="gpt2")
+generator = pipeline("text-generation", model="microsoft/BioGPT")
 summarizer = load_summarizer()
+translator = pipeline("translation", model="Helsinki-NLP/opus-mt-any-de")
 
 image_file = st.sidebar.file_uploader("Upload an Image (.jpg, .jpeg, .png)", type=["jpg", "jpeg", "png"])
 # Create tabs for each functionality
@@ -210,7 +211,8 @@ with tab3:
         # translator = GoogleTranslator(source='en', target='de')
         # german_translation = translator.translate(medical_terms)
         # Translate the text
-        translation = translate_text(medical_terms, target='de')
+        # translation = translate_text(medical_terms, target='de')
+        translation = translator(medical_terms)
         outputs = 'Unable to translate, please Retry ...'
         if translation:
             outputs = translation