Update README.md

README.md

@@ -17,18 +17,18 @@ import torch
 import requests
 from PIL import Image
 from transformers import BlipProcessor, BlipForConditionalGeneration
+import re
 
 def prepare(text):
     text = text.replace('. ','.').replace(' .','.')
-    text = text.replace('< ','<').replace(' <','<')
-    text = text.replace('> ','>').replace(' >','>')
     text = text.replace('( ','(').replace(' (','(')
     text = text.replace(') ',')').replace(' )',')')
     text = text.replace(': ',':').replace(' :',':')
     text = text.replace('_ ','_').replace(' _','_')
-    text = text.replace(',(())','')
+    text = text.replace(',(())','').replace('(()),','')
     for i in range(10):
         text = text.replace(')))','))').replace('(((','((')
+    text = re.sub(r'<[^>]*>', '', text)
     return text
 
 path_to_model = "blip-image2promt-stable-diffusion-v0.15"
@@ -46,7 +46,7 @@ out = model.generate(**inputs, max_new_tokens=100)
 
 out_txt = processor.decode(out[0], skip_special_tokens=True)
 
-print(prepare(out_txt))
+print(prepare(out_txt)) # woman sitting on the beach at sunset, rear view,((happy)),((happy)),((dog)),((mixed)),(()),((
 ```