kingmadhu1 commited on
Commit
338367b
·
verified ·
1 Parent(s): c407d8d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=100)
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(share=True)
 
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()