init

README.md

@@ -1,6 +1,6 @@
 ---
 title: Guess The Image
-emoji: 📚
+emoji: 🤗
 colorFrom: pink
 colorTo: gray
 sdk: streamlit

main.py

@@ -0,0 +1,68 @@
+from diffusers import AutoPipelineForText2Image
+import torch
+import gradio as gr
+import numpy as np
+from openai import OpenAI
+import os
+client = OpenAI()
+
+import streamlit as st
+from PIL import Image
+
+@st.cache_data(ttl=3600)
+def get_prompt_to_guess():
+    response = client.chat.completions.create(
+    model="gpt-3.5-turbo",
+    messages=[
+        {"role": "system", "content": "You are a helpful assistant to generate one simple prompt in order to generate an image. Your given prompt won't go over 10 words. You only return the prompt. You will also answer in french."},
+        {"role": "user", "content": "Donne moi un prompt pour generer une image"},
+    ]
+    )
+    return response.choices[0].message.content
+
+@st.cache_resource
+def get_model():
+    pipe = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float32, variant="fp16")
+    return pipe
+
+@st.cache_data
+def generate_image(_pipe, prompt):
+    return _pipe(prompt=prompt, num_inference_steps=1, guidance_scale=0.0, seed=1).images[0]
+
+if "submit_guess" not in st.session_state:
+    st.session_state["submit_guess"] = False
+
+def check_prompt(prompt, prompt_to_guess):
+    return prompt.strip() == prompt_to_guess.strip()
+
+pipe = get_model()
+prompt = get_prompt_to_guess()
+im_to_guess = generate_image(pipe, prompt)
+h, w = im_to_guess.size
+
+st.title("Guess the prompt by Ekimetrics")
+st.text("Rules : guess the prompt (in French) to generate the left image with the sdxl turbo model")
+st.text("Hint : use right side to help you guess the prompt by testing some")
+st.text("Disclosure : this runs on CPU so generation are quite slow (even with sdxl turbo)")
+col_1, col_2 = st.columns([0.5, 0.5])
+with col_1:
+    st.header("GUESS THE PROMPT")
+    st.image(im_to_guess)
+    guessed_prompt = st.text_area("Input your guess prompt")
+    st.session_state["submit_guess"] = st.button("guess the prompt")
+    if st.session_state["submit_guess"]:
+        if check_prompt(guessed_prompt, prompt):
+            st.text("Good prompt ! test again in 24h !")
+        else:
+            st.text("wrong prompt !")
+
+with col_2:
+    st.header("TEST THE PROMPT")
+    testing_prompt = st.text_area("Input your testing prompt")
+    st.session_state["testing"] = st.button("test the prompt")
+    if st.session_state["testing"]:
+        im = generate_image(pipe, testing_prompt)
+        st.session_state["testing"] = False
+    else:
+        im = np.zeros([h,w,3])
+    st.image(im)

requirements.txt

@@ -0,0 +1,4 @@
+diffusers
+transformers
+accelerate
+streamlit