interface_more

app.py

@@ -30,10 +30,25 @@ def classify_image(inp):
   return confidences
 
 
-gr.Interface(fn=classify_image, 
-             inputs=gr.Image(shape=(224, 224)),
-             outputs=gr.Label(num_top_classes = 2),
+image_interface = gr.Interface(fn=classify_image, 
+             inputs=gr.inputs.Image(shape=(224, 224)),
+             outputs=gr.outputs.Label(num_top_classes = 2),
              title="Demo",
              description="Here's a sample image classification. Enjoy!",
-             examples=[['300104.png']]
-             ).launch()
+             examples=[['300104.png']])
+
+image_interface.launch()
+
+text_interface = pipeline("text-generation", model="densenet")
+
+def generate_text(prompt):
+  text = text_interface(prompt, max_length=50)[0]['generated_text']
+  return text
+
+text_generator_interface = gr.Interface(fn=generate_text,
+                                         inputs="text",
+                                         outputs="text",
+                                         title="Text Generator",
+                                         description="Generate text using GPT-2",
+                                         examples=[["300104.png"]])
+text_generator_interface.launch()