stchakman commited on
Commit
15c7520
·
1 Parent(s): dbd05db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ def generate_dishes(ingredients, n=3, max_tokens=150, temperature=0.7):
52
  return dishes
53
 
54
  def generate_image(prompt):
55
- model_id = "stabilityai/stable-diffusion-2-1"
56
  pipe = StableDiffusionPipeline.from_pretrained(model_id)
57
 
58
  # If you have a GPU available, uncomment the following line
 
52
  return dishes
53
 
54
  def generate_image(prompt):
55
+ model_id = "runwayml/stable-diffusion-v1-5"
56
  pipe = StableDiffusionPipeline.from_pretrained(model_id)
57
 
58
  # If you have a GPU available, uncomment the following line