File size: 4,134 Bytes
6f3b453 10e69ed 6f3b453 10e69ed 2843e59 10e69ed 2843e59 10e69ed 99c153c 10e69ed 0a5d7f7 10e69ed 6f3b453 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 |
import gradio as gr
import requests
from image_utils import print_text_on_image_centered, create_background_image
from hf_utils import hf_validate_api_token
from segmentation_utils import segment_and_overlay_results
def segment_gradio_image(api_token, model, image):
# Validacion del token y la imagen
is_token_valid, api_token_message = hf_validate_api_token(api_token)
if not is_token_valid:
text_image = print_text_on_image_centered(
create_background_image(500, 500, "white"),
'HuggingFace API Token invalid. Please enter a valid token.',
'red'
)
segments_list = "No segments available."
return api_token_message, text_image, segments_list
else:
if image is None:
text_image = print_text_on_image_centered(
create_background_image(500, 500, "white"),
'No image detected',
'orange'
)
segments_list = "No segments available."
return api_token_message, text_image, segments_list
else:
text_image = print_text_on_image_centered(
create_background_image(500, 500, "white"),
'PROCESANDO',
'blue'
)
segments_list = "No segments available."
# Assuming segment_image is a placeholder for actual segmentation function
# Uncomment and modify this part according to your segmentation implementation
# response = segment_image(api_token, model, image)
# text_image = response["segmented_image"]
text_image, segments = segment_and_overlay_results(image,api_token, model)
print("app.py segment_gradio_image")
segments_list = "Segments:\n"
for segment in segments:
print(segment['label'] + " " + str(segment['score']))
segments_list += f"\n{segment['label']}: {segment['score']}"
return api_token_message, text_image, segments_list
with gr.Blocks() as demo:
gr.Markdown("# Manu Panoptic Segmentation demo with Hugging Face v0.1 🤗")
gr.Markdown("You can use the default model or enter your own model.")
gr.Markdown("You can use the default image or upload your own image.")
gr.Markdown("You can get your API token from your Hugging Face account.")
gr.Markdown("You can find more models at https://huggingface.co/models")
gr.Markdown("You can find more images at https://unsplash.com/")
with gr.Row():
api_token = gr.Textbox(
label="API Token",
placeholder="Enter your Hugging Face API token here"
)
model_name = gr.Textbox(
label="AI Segmentation Model",
placeholder="Enter your Segmentation model here",
value="facebook/mask2former-swin-tiny-coco-panoptic"
)
image_input = gr.Image(label="Upload Image")
with gr.Row():
api_token_validation = gr.Textbox(label="API Token Validation")
segmented_image = gr.Image(label="Segmented Image")
# New block for segments output
with gr.Row():
segments_output = gr.Textbox(label="Segments")
examples = gr.Examples(
examples=[
["Your HF API Token", "facebook/mask2former-swin-tiny-coco-panoptic", "https://upload.wikimedia.org/wikipedia/commons/7/74/A-Cat.jpg"]
],
inputs=[api_token, model_name, image_input]
)
api_token.change(
fn=segment_gradio_image,
inputs=[api_token, model_name, image_input],
outputs=[api_token_validation, segmented_image, segments_output]
)
model_name.change(
fn=segment_gradio_image,
inputs=[api_token, model_name, image_input],
outputs=[api_token_validation, segmented_image, segments_output]
)
image_input.change(
fn=segment_gradio_image,
inputs=[api_token, model_name, image_input],
outputs=[api_token_validation, segmented_image, segments_output]
)
demo.launch() |