Update app.py

app.py

@@ -20,7 +20,8 @@ API_URL = "https://api-inference.huggingface.co/models/bigscience/bloom"
 
 def translate(prompt_ , to_lang): 
   input_prompt = "translate this"
-  seed = 42
+  seed = 42 
+  
   prompt =  f"Instruction : Given an English input sentence translate it into {to_lang} sentence. \n input : \"{prompt_}\" \n {to_lang} : " 
   if len(prompt) == 0:
     prompt = input_prompt 
@@ -153,8 +154,11 @@ def predict(cap_col, img, target_language):
     pred_caption = image_to_text(img)
 
     cap_col.markdown('#### Predicted Captions:')
-    translated_caption = translate(pred_caption, target_language)
-    captions.append(translated_caption)
+    if target_language == 'English':
+        captions.append(pred_caption)
+    else:
+       translated_caption = translate(pred_caption, target_language)
+       captions.append(translated_caption)
 
    # for _ in range(4):
         #pred_caption = image_to_text(img)
@@ -179,7 +183,7 @@ def generate_caption_section():
     # Language selection dropdown
            
                           
-    target_language = st.selectbox('Select Target Language', ['Tamil', 'Spanish', 'Hindi', 'Italian'], index=0)
+    target_language = st.selectbox('Select Target Language', ['English', 'Spanish', 'Hindi', 'Italian'], index=0)
 
     # Process image and generate captions
     if img_url: