Update app.py

app.py

@@ -10,10 +10,10 @@ def chatgpt(query):
     return dictionary[query]
 
 def generateimage1(query):
-    return query.replace(' ', '-') + '.png'
+    return os.path.join(os.path.dirname(__file__), query.replace(' ', '-') + '.png')
 
 def generateimage2(query):
-    return query.replace(' ', '-') + '-chatgpt.png'
+    return os.path.join(os.path.dirname(__file__), query.replace(' ', '-') + '-chatgpt.png')
 
 with gr.Blocks() as demo:
     gr.Markdown("Ask **ChatGPT** to generate detailed descriptbions.")