potatopizza commited on
Commit
10ab55c
·
verified ·
1 Parent(s): 5f72a3b

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +2 -2
inference.py CHANGED
@@ -49,10 +49,10 @@ def inference(
49
 
50
  if __name__ == "__main__":
51
  # Example prompt
52
- sample_prompt = "A painting of the Eiffel Tower in the style of Eric Fischl."
53
 
54
  # Assuming the fine-tuned model was saved with pipe.save_pretrained("stable-diffusion-wikiart-final")
55
- finetuned_model_path = "/home/work/daehyun/Style-Portrait-Generator/scripts/stable-diffusion-wikiart-final"
56
 
57
  # Run inference
58
  inference(
 
49
 
50
  if __name__ == "__main__":
51
  # Example prompt
52
+ sample_prompt = "A paint of Eiffel tower, in the style of eric fischl."
53
 
54
  # Assuming the fine-tuned model was saved with pipe.save_pretrained("stable-diffusion-wikiart-final")
55
+ finetuned_model_path = "stable-diffusion-wikiart-final"
56
 
57
  # Run inference
58
  inference(