Text-to-Image / app.py
Jonny001's picture
Create app.py
b53c295 verified
raw
history blame
596 Bytes
import gradio as gr
from transformers import pipeline
def face_projection(text):
processed_text = preprocess_text(text)
face_image = generate_face_image(processed_text)
return face_image
def preprocess_text(text):
return text
def generate_face_image(processed_text):
face_projection_pipeline = pipeline("text2image", model="Purz/face-projection")
face_image = face_projection_pipeline(processed_text)
return face_image
iface = gr.Interface(
fn=face_projection,
inputs="text",
outputs="image",
title="Face Projection from Text"
)
iface.launch()