Update app.py

app.py

@@ -3,6 +3,7 @@ import os
 import json 
 import requests
 import openai
+import re
 
 #Streaming endpoint 
 API_URL = "https://api.openai.com/v1/chat/completions" 
@@ -17,7 +18,8 @@ stable_diffusion = gr.Blocks.load(name="spaces/runwayml/stable-diffusion-v1-5")
 
 def get_images(username):
     prompt_string = query(username)
-    prompt_list = [element.split(". ")[1] for element in prompt_string.split(", ")]
+    regexed_prompt_string = re.sub(r'@\w+', '', prompt_string)
+    prompt_list = [element.split(". ")[1] for element in regexed_prompt_string.split(", ")]
     prompt_list_prepend = [f"an photo of {item}, photorealistic, coherent, high quality, rendered in unreal engine" for item in prompt_list]
     print(prompt_list_prepend)