add option for rearrange prompt

app.py

@@ -19,7 +19,7 @@ model = Magma.from_checkpoint(
     device = 'cuda:0'
 )
 
-def generate(image,context, length, temperature, top_k):
+def generate(image,context, length, temperature, top_k,rearrange):
   # context = context.strip()
 
   # url_regex = r'https?:\/\/(www\.)?[-a-zA-Z0-9@:%._\+~#=]{1,256}\.[a-zA-Z0-9()]{1,6}\b([-a-zA-Z0-9()@:%_\+.~#?&//=]*)'
@@ -33,12 +33,18 @@ def generate(image,context, length, temperature, top_k):
   #       return str(e)
   #   else:
   #     inputs.append(line)
-
-  inputs =[
-    ## supports urls and path/to/image
-    ImageInput(image),
-    context
-  ]
+  if rearrange:
+    inputs =[
+      ## supports urls and path/to/image
+      context,
+      ImageInput(image)
+    ]
+  else:
+    inputs =[
+      ## supports urls and path/to/image
+      ImageInput(image),
+      context
+    ]
 
   ## returns a tensor of shape: (1, 149, 4096)
   embeddings = model.preprocess_inputs(inputs)  
@@ -53,7 +59,7 @@ def generate(image,context, length, temperature, top_k):
 
   return output[0]
   
-examples=[["woods_hi.jpeg","Describe the painting:",15,0.7,0]]
+examples=[["woods_hi.jpeg","Describe the painting:",15,0.7,0,False]]
 
 title="MAGMA"
 description="MAGMA -- Multimodal Augmentation of Generative Models through Adapter-based Finetuning"
@@ -61,13 +67,14 @@ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2112.052
 iface = gr.Interface(
   fn=generate,
   inputs=[
-    gr.inputs.Image(type="filepath"),gr.inputs.Textbox(
-      label="Prompt:",
+    gr.inputs.Image(type="filepath",label="Image Prompt"),gr.inputs.Textbox(
+      label="Text Prompt:",
       default="Describe the painting:",
       lines=7),
     gr.inputs.Slider(minimum=1, maximum=100, default=15, step=1, label="Output tokens:"),
     gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.7, label='Temperature'),
-    gr.inputs.Slider(minimum=0, maximum=100, default=0, step=1, label='Top K')
+    gr.inputs.Slider(minimum=0, maximum=100, default=0, step=1, label='Top K'),
+    gr.inputs.Checkbox(default=False, label="Rearrange Prompt", optional=False)
   ],
   outputs=["textbox"],
   examples=examples,