Update app.py

app.py

@@ -43,7 +43,7 @@ def gemini_predict(prompt):
                         Query : {prompt}
                         """
     response = gemini_flash.generate_content(system_message)
-    return(str(response.text)[-2])
+    return(str(response.text)[:-2])
 
 
 MAX_SEED = np.iinfo(np.int32).max
@@ -63,6 +63,7 @@ def process(input_image_editor, mask_image, input_text, strength, seed, randomiz
     if not input_text:
         raise gr.Error("Please enter a text prompt.")
     item = gemini_predict(input_text)
+    #print(item)
     
     image = input_image_editor['background']
     if not image:
@@ -81,7 +82,7 @@ def process(input_image_editor, mask_image, input_text, strength, seed, randomiz
     
     return result, mask_image, seed, item
 
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
+with gr.Blocks(theme='upsatwal/mlsc_tiet') as demo:
     gr.Markdown(MARKDOWN)
     with gr.Row():
         with gr.Column(scale=1):
@@ -129,8 +130,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
             with gr.Accordion("Upload a mask", open=False):
                 uploaded_mask_component = gr.Image(label="Already made mask (black pixels will be preserved, white pixels will be redrawn)", sources=["upload"], type="pil")
-            submit_button_component = gr.Button(
-                value='Inpaint', variant='primary')
+            submit_button_component = gr.Button(value='Inpaint', variant='primary')
         with gr.Column(scale=1):
             output_image_component = gr.Image(type='pil', image_mode='RGB', label='Generated Image')
             output_mask_component = gr.Image(type='pil', image_mode='RGB', label='Generated Mask')