Adding code to download baseline model.

app.py

@@ -23,6 +23,9 @@ from transformers import pipeline
 
 model_id = "niclasfw/bloom-1b5-schlager-bot-004"
 
+model = AutoModelForCausalLM.from_pretrained(model_id)
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+
 generator = pipeline(task="text-generation", model=model_id, tokenizer=model_id)
 
 st.title('Schlager Bot')