alomari7 commited on
Commit
fca94bd
·
verified ·
1 Parent(s): 120deb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -35,7 +35,8 @@ try:
35
  print(f"✅ Trained model '{MODEL_ID}' loaded successfully.")
36
  except Exception:
37
  print(f"❌ Could not load trained model '{MODEL_ID}'. Loading base model.")
38
- pipe = DiffusionPipeline.from_pretrained(BASE_MODEL_ID, torch_dtype=DTYPE, safety_checker=None)
 
39
 
40
  pipe = pipe.to(DEVICE)
41
 
 
35
  print(f"✅ Trained model '{MODEL_ID}' loaded successfully.")
36
  except Exception:
37
  print(f"❌ Could not load trained model '{MODEL_ID}'. Loading base model.")
38
+ pipe = DiffusionPipeline.from_pretrained
39
+ (BASE_MODEL_ID, torch_dtype=DTYPE, safety_checker=None)
40
 
41
  pipe = pipe.to(DEVICE)
42