mirror of
https://github.com/gradio-app/gradio.git
synced 2024-12-03 01:50:59 +08:00
38 lines
817 B
Python
38 lines
817 B
Python
# This demo needs to be run from the repo folder.
|
|
# python demo/fake_gan/run.py
|
|
import os
|
|
import random
|
|
import time
|
|
|
|
import gradio as gr
|
|
|
|
|
|
def fake_gan(desc):
|
|
if desc == "NSFW":
|
|
raise gr.Error("NSFW - banned content.")
|
|
if desc == "error":
|
|
raise ValueError("error")
|
|
time.sleep(2)
|
|
image = random.choice(
|
|
[
|
|
"files/cheetah1.jpg",
|
|
"files/elephant.jpg",
|
|
"files/tiger.jpg",
|
|
"files/zebra.jpg",
|
|
]
|
|
)
|
|
return image
|
|
|
|
|
|
demo = gr.Interface(
|
|
fn=fake_gan,
|
|
inputs=gr.Textbox(),
|
|
outputs=gr.Image(label="Generated Image"),
|
|
title="FD-GAN",
|
|
description="This is a fake demo of a GAN. In reality, the images are randomly chosen from Unsplash.",
|
|
)
|
|
demo.queue(max_size=3)
|
|
|
|
if __name__ == "__main__":
|
|
demo.launch()
|