jeremyLE-Ekimetrics commited on
Commit
6804529
1 Parent(s): 7fb418b
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -63,7 +63,7 @@ def get_prompt_to_guess(index):
63
  {"role": "user", "content": f"Donne moi un prompt pour generer une image de {random.choice(random_prompt)}"},
64
  ]
65
  )
66
- return response.choices[0].message.content
67
 
68
  @st.cache_resource
69
  def get_model():
 
63
  {"role": "user", "content": f"Donne moi un prompt pour generer une image de {random.choice(random_prompt)}"},
64
  ]
65
  )
66
+ return response.choices[0].message.content.replace('"','').replace("'","")
67
 
68
  @st.cache_resource
69
  def get_model():