potatopizza
commited on
Update inference.py
Browse files- 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
|
53 |
|
54 |
# Assuming the fine-tuned model was saved with pipe.save_pretrained("stable-diffusion-wikiart-final")
|
55 |
-
finetuned_model_path = "
|
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(
|