Try relative path

app.py

@@ -53,7 +53,7 @@ def upload_image(state, image_input):
     input_image = Image.open(image_input.name).resize(
         (224, 224)).convert('RGB')
     input_image.save(image_input.name)  # Overwrite with smaller image.
-    conversation += [(f'<img src="/file={image_input.name}" style="display: inline-block;">', "")]
+    conversation += [(f'<img src="./file={image_input.name}" style="display: inline-block;">', "")]
     return [conversation, chat_history + [input_image, ""]], conversation
 
 
@@ -116,11 +116,11 @@ def generate_for_prompt(input_text, state, ret_scale_factor, num_words, temperat
             if p['decision'] is not None and p['decision'][0] == 'gen':
                 image = p['gen'][0][0].resize((224, 224))
                 filename = save_image_to_local(image)
-                response += f'<img src="/file={filename}" style="display: inline-block;"><p style="font-size: 12px; color: #555;">(Generated)</p>'
+                response += f'<img src="./file={filename}" style="display: inline-block;"><p style="font-size: 12px; color: #555;">(Generated)</p>'
             else:
                 image = p['ret'][0][0].resize((224, 224))
                 filename = save_image_to_local(image)
-                response += f'<img src="/file={filename}" style="display: inline-block;"><p style="font-size: 12px; color: #555;">(Retrieved)</p>'
+                response += f'<img src="./file={filename}" style="display: inline-block;"><p style="font-size: 12px; color: #555;">(Retrieved)</p>'
 
 
     chat_history = model_inputs + \