CoolFace
Apppublic

codingnplayinjeonju/funny

sourceHugging Faceapache-2.0updated 1y agoView on Hugging Face
0likes
generateText.py35 linesDownload Raw Back to root
1from openai import OpenAI2from dotenv import load_dotenv3import os4from PIL import Image5from ImageProcess import classify_image6load_dotenv()7api_key = os.environ.get("OPENAI_API_KEY")  # 실제 키 입력8 9def create_funny_caption(label):10    client = OpenAI(11        api_key=api_key,12    )13    print(f"label:{label}")14    prompt = f"""이 이미지는 label: "{label}"이 표시된 이미지인데,15        resnet50 모델이 분류한 결과야.16        이 Label 과 관련된 기발하고 재치있는 농담을 해 줬으면 좋겠어.17        """18    print("prompt:", prompt)19    response = client.responses.create(20        model="gpt-3.5-turbo",21        instructions="""22        너는 밈 아이디어를 생성하는 창의적 코미디 작가야.23        니가 이미지를 직접 보지 못한다는 것은 알고 있어. 24        그러니 이미지를 직접 보지 못한다는 말은 하지 말아줘.25        대신 이미지를 설명하는 prompt 를 이해한 뒤, 짧고 재밌는 문구를 만들어줘.26        문구는 100자 이내로 해줘.27        문구는 한국어로 작성해줘.28        분석라벨 : 뒤에 오는 라벨들도 한국어로 번역해서 보여주면 더 좋겠어.29        """,30        input=prompt31    )32    return response.output_text33 34 35