Minor change

Zain-ul-din · Oct 24, 2024 · 6b64895 · 6b64895
1 parent f6aeddb
commit 6b64895
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/models/OpenAIModel.ts b/src/models/OpenAIModel.ts
@@ -64,7 +64,7 @@ class ChatGPTModel extends AIModel<AIArguments, AIHandle> {
     const image: Image = res.data[0];
     const base64Img = Buffer.from(image.b64_json as string, 'base64');
 
-    return { image: base64Img, caption: image.revised_prompt || '' };
+    return { image: base64Img, caption: image.revised_prompt || prompt };
   }
 
   public async sendMessage({ sender, prompt, prefix }: AIArguments, handle: AIHandle) {