Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ text_model.to(device)
|
|
13 |
|
14 |
def generate_text(prompt):
|
15 |
inputs = tokenizer(prompt, return_tensors="pt").to(device)
|
16 |
-
outputs = text_model.generate(**inputs, max_length=
|
17 |
return tokenizer.decode(outputs[0], skip_special_tokens=True)
|
18 |
|
19 |
# Load Image Generation Model (Stable Diffusion)
|
@@ -58,4 +58,4 @@ with gr.Blocks(title="Charriot_v2") as demo:
|
|
58 |
audio_btn = gr.Button("Generate Audio")
|
59 |
audio_btn.click(fn=generate_audio, inputs=audio_prompt, outputs=audio_output)
|
60 |
|
61 |
-
demo.launch(
|
|
|
13 |
|
14 |
def generate_text(prompt):
|
15 |
inputs = tokenizer(prompt, return_tensors="pt").to(device)
|
16 |
+
outputs = text_model.generate(**inputs, max_length=10000)
|
17 |
return tokenizer.decode(outputs[0], skip_special_tokens=True)
|
18 |
|
19 |
# Load Image Generation Model (Stable Diffusion)
|
|
|
58 |
audio_btn = gr.Button("Generate Audio")
|
59 |
audio_btn.click(fn=generate_audio, inputs=audio_prompt, outputs=audio_output)
|
60 |
|
61 |
+
demo.launch()
|