Update app.py

app.py

@@ -17,6 +17,7 @@ def img2txt(url):
     print("Generating text from the image...")
     text = captioning_model(url, max_new_tokens=20)[0]["generated_text"]
     
+    print(text)
     print("Text generated successfully.")
     return text
 
@@ -38,7 +39,7 @@ def txt2story(img_text):
     messages = [
         {"role": "system", "content": '''You are an expert short story writer who specialises 
         in writing stories on input set of words.''', "temperature": 0.8},
-        {"role": "user", "content": "f'Here is input set of words: '{img_text}''", "temperature": 1},
+        {"role": "user", "content": "Here is input set of words: {img_text}", "temperature": 1},
     ]
     
     print("Generating story completion using the AI model...")
@@ -49,6 +50,7 @@ def txt2story(img_text):
 
     print("Story generated successfully.")
     return chat_completion.choices[0].message.content
+    print(chat_completion.choices[0].message.content)
 
 
 # Text-to-speech