Update app.py
Browse files
app.py
CHANGED
@@ -1,17 +1,7 @@
|
|
1 |
import gradio as gr
|
2 |
-
import
|
3 |
-
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
4 |
|
5 |
-
|
6 |
-
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|
7 |
-
model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
|
8 |
-
|
9 |
-
def generate_images(description):
|
10 |
-
input_ids = tokenizer.encode(description, return_tensors="pt")
|
11 |
-
# Model generates a batch of one image
|
12 |
-
output = model.generate(input_ids)
|
13 |
-
output_image = output[0].numpy().transpose(1,2,0)
|
14 |
-
return output_image.astype("uint8")
|
15 |
|
16 |
inputs = gr.inputs.Textbox(prompt="Enter Text Description")
|
17 |
outputs = gr.outputs.Image(label="Generated Image")
|
|
|
1 |
import gradio as gr
|
2 |
+
from diffusers import DiffusionPipeline
|
|
|
3 |
|
4 |
+
pipeline = DiffusionPipeline.from_pretrained("cloudqi/cqi_text_to_image_pt_v0")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
5 |
|
6 |
inputs = gr.inputs.Textbox(prompt="Enter Text Description")
|
7 |
outputs = gr.outputs.Image(label="Generated Image")
|