mikegarts commited on
Commit
4efa0ed
1 Parent(s): 5578813

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -44,7 +44,7 @@ def generate(prompt):
44
 
45
  def make_image(prompt):
46
  pipe = get_image_pipe()
47
- image = pipe(prompt).images[0]
48
 
49
  def predict(prompt):
50
  story = generate(prompt=prompt)
@@ -63,4 +63,4 @@ gr.Interface(
63
  title=title,
64
  description=description,
65
  examples=[["My new adventure would be"], ["Then I a hobbit appeared"], ["Frodo told me"]]
66
- ).launch(share=True)
 
44
 
45
  def make_image(prompt):
46
  pipe = get_image_pipe()
47
+ image = pipe(prompt)["sample"][0]
48
 
49
  def predict(prompt):
50
  story = generate(prompt=prompt)
 
63
  title=title,
64
  description=description,
65
  examples=[["My new adventure would be"], ["Then I a hobbit appeared"], ["Frodo told me"]]
66
+ ).launch(debug=True)