Update app.py

app.py

@@ -1,15 +1,14 @@
-#import gradio as gr
-#gr.Interface.load("models/teo-sanchez/en_ner_prompting").launch()
 import gradio as gr
 
-from transformers import pipeline
-
-pipe = pipeline("ner", model="models/teo-sanchez/en_ner_prompting")
-
-iface = gr.Interface(
-  inputs='text',
-  outputs='text',
-  examples=[["Hyperrealistic dslr film still of! justin bieber! disguised as beaver, stunning 8 k octane comprehensive 3 d render, inspired by istvan sandorfi & greg rutkowski & unreal engine, perfect symmetry, dim volumetric cinematic lighting, extremely hyper - detailed, incredibly real lifelike attributes & flesh texture, intricate, masterpiece, artstation, stunning"]]
-)
-
-iface.launch()
+iface = gr.Interface.load("models/teo-sanchez/en_ner_prompting",
+                          title = 'Prompt entity recognizer by Selas.ai',
+                          description="Recognize different entities within an AI prompt",
+                          examples=[["Hyperrealistic dslr film still of a cute calico cat, stunning 8 k octane comprehensive 3 d render, inspired by istvan sandorfi & greg rutkowski & unreal engine, perfect symmetry, dim volumetric cinematic lighting, extremely hyper - detailed, incredibly real lifelike attributes & flesh texture, intricate, masterpiece, artstation, stunning"],
+                                    ["Quetzalcoatl in an epic battle with garuda, fantasy, stained glass, d & d, intricate, elegant, highly detailed, digital painting, artstation, concept art, matte, sharp focus, illustration, art by john collier and albert aublet and krenz cushart and artem demura and alphonse mucha"]],
+                          live = True,
+                          )
+
+iface.input_components[0].label = "Copy-paste an AI prompt you'd like to analyze"
+iface.input_components[0].value = "Hyperrealistic dslr film still of! justin bieber! disguised as beaver, stunning 8 k octane comprehensive 3 d render, inspired by istvan sandorfi & greg rutkowski & unreal engine, perfect symmetry, dim volumetric cinematic lighting, extremely hyper - detailed, incredibly real lifelike attributes & flesh texture, intricate, masterpiece, artstation, stunning"
+iface.output_components[0].label = "Entity recognized in the prompt"
+iface.launch()