Spaces:
Running
Running
jeremyLE-Ekimetrics
commited on
Commit
•
6804529
1
Parent(s):
7fb418b
fix
Browse files
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():
|