Create app.py

app.py

@@ -0,0 +1,44 @@
+import gradio as gr
+import torch
+from transformers import pipeline
+
+# Load the models
+caption_generator = pipeline("image-captioning", model="gokaygokay/SD3-Long-Captioner")
+compliment_generator = pipeline("text-generation", model="hysts/zephyr-7b")
+
+SYSTEM_PROMPT = """
+You are helpful assistant that gives the best compliments to people.
+You will be given a caption of someone's headshot.
+Based on that caption, provide a one sentence compliment to the person in the image.
+Make sure you compliment the person in the image and not any objects or scenery.
+Do NOT include any hashtags in your compliment or phrases like (emojis: dog, smiling face with heart-eyes, sun).
+Here are some examples of the desired behavior:
+Caption: a front view of a man who is smiling, there is a lighthouse in the background, there is a grassy area on the left that is green and curved. in the distance you can see the ocean and the shore. there is a grey and cloudy sky above the lighthouse and the trees.
+Compliment: Your smile is as bright as a lighthouse, lighting up the world around you. 🌟
+Caption: in a close-up, a blonde woman with short, wavy hair, is the focal point of the image. she's dressed in a dark brown turtleneck sweater, paired with a black hat and a black suit jacket. her lips are a vibrant red, and her eyes are a deep brown. in the background, a man with a black hat and a white shirt is visible.
+Compliment: You are the epitome of elegance and grace, with a style that is as timeless as your beauty. 💃🎩
+Conversation begins below:
+"""
+
+def generate_compliment(image):
+    # Generate caption
+    caption = caption_generator(image)[0]['caption']
+    # Generate compliment
+    input_text = f"Caption: {caption}\nCompliment: "
+    generated_compliment = compliment_generator(SYSTEM_PROMPT + input_text, max_new_tokens=50, do_sample=True)[0]['generated_text']
+    compliment = generated_compliment.split('Compliment: ')[-1].strip()
+    return caption, compliment
+
+# Gradio interface
+iface = gr.Interface(
+    fn=generate_compliment,
+    inputs=gr.inputs.Image(type="pil"),
+    outputs=[
+        gr.outputs.Textbox(label="Caption"),
+        gr.outputs.Textbox(label="Compliment")
+    ],
+    title="Compliment Bot 💖",
+    description="Upload your headshot and get a personalized compliment!"
+)
+
+iface.launch()