Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+from gradio_client.client import DEFAULT_TEMP_DIR
 from huggingface_hub import InferenceClient
 import json
 import random
@@ -32,6 +33,9 @@ def array_random(array):
  if not array: return None
  return random.choice(array)
  
+def add_file(file):
+ return file.name, gr.update(label='🖼️ Uploaded!')
+ 
 def predict(access_key, tokens, instructions, image, pre_input, input, seed = 0): 
  
  if (access_key != KEY):
@@ -42,7 +46,7 @@ def predict(access_key, tokens, instructions, image, pre_input, input, seed = 0)
  
  token = array_random(string_to_json(tokens))
 
- formatted_input = instructions.replace(symbol, pre_input) + pre_input + input + " " + image
+ formatted_input = instructions.replace(symbol, pre_input) + pre_input + input + "![](" + image + ")"
 
  print(formatted_input)
  
@@ -68,7 +72,8 @@ def maintain_cloud():
 with gr.Blocks() as demo:
  with gr.Row():
  with gr.Column():
- image = gr.Image()
+ image = gr.Image(type="filepath", label="Image Input", visible=True)
+ upload = gr.UploadButton("Upload Image", file_types=["image"])
  instructions = gr.Textbox(label="Instruction", placeholder="Message...", value=default_instructions, lines=1)
  pre_text = gr.Textbox(label="Pre-Input", placeholder="Message...", value=default_pre_text, lines=1)
  text = gr.Textbox(label="Text Input", placeholder="Message...", lines=2)
@@ -81,6 +86,7 @@ with gr.Blocks() as demo:
  run = gr.Button("Generate")
  cloud = gr.Button("☁️")
 
+ upload.upload(add_file, [upload], [image], queue = False)
  run.click(predict, inputs=[access_key, json_tokens, instructions, image, pre_text, text, seed], outputs=[output], queue = False)
  cloud.click(maintain_cloud, inputs = [], outputs = [output], queue = False)