Donw

app.py

@@ -6,7 +6,7 @@ from PIL import Image
 import torch
 from transformers import AutoProcessor, LlavaForConditionalGeneration, BitsAndBytesConfig
 from deep_translator import GoogleTranslator
-from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
+# from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
 import warnings 
 from flask import Flask
 
@@ -93,26 +93,26 @@ def facebook_bn_en(input_sentence):
   return translated_text_en
 # print("Translated English:", translated_text_en)
 
-def facebook_en_bn(input_sentence):
-  # Translate English to Bengali
-#   model = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_418M")
-#   tokenizer = M2M100Tokenizer.from_pretrained("facebook/m2m100_418M")
-  tokenizer.src_lang = "en"
-  encoded_en = tokenizer(input_sentence, return_tensors="pt")
-  generated_tokens = model.generate(**encoded_en, forced_bos_token_id=tokenizer.get_lang_id("bn"))
-  translated_text_bn = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
-  return translated_text_bn
-
-def facebook_response(url, input_sentence):
-  url = input("ইমেজ url লিখুন: ")
-  input_sentence = input("ছবি সম্পর্কে আপনার প্রশ্ন লিখুন: ")
-  image_prompt = facebook_bn_en(input_sentence)
-  response = inference(image_prompt, url)
-  assistant_index = response.find("ASSISTANT:")
-  extracted_string = response[assistant_index + len("ASSISTANT:"):].strip()
-  output = facebook_en_bn(extracted_string)
-  print("বটী: ", output)
-  return output
+# def facebook_en_bn(input_sentence):
+#   # Translate English to Bengali
+# #   model = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_418M")
+# #   tokenizer = M2M100Tokenizer.from_pretrained("facebook/m2m100_418M")
+#   tokenizer.src_lang = "en"
+#   encoded_en = tokenizer(input_sentence, return_tensors="pt")
+#   generated_tokens = model.generate(**encoded_en, forced_bos_token_id=tokenizer.get_lang_id("bn"))
+#   translated_text_bn = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
+#   return translated_text_bn
+
+# def facebook_response(url, input_sentence):
+#   url = input("ইমেজ url লিখুন: ")
+#   input_sentence = input("ছবি সম্পর্কে আপনার প্রশ্ন লিখুন: ")
+#   image_prompt = facebook_bn_en(input_sentence)
+#   response = inference(image_prompt, url)
+#   assistant_index = response.find("ASSISTANT:")
+#   extracted_string = response[assistant_index + len("ASSISTANT:"):].strip()
+#   output = facebook_en_bn(extracted_string)
+#   print("বটী: ", output)
+#   return output
 
 
 image_cache = {}