Update app.py

app.py

@@ -37,6 +37,7 @@ model = AutoModelForCausalLM.from_pretrained(
     low_cpu_mem_usage=True
 )
 
+print( "initalized model")
 # model_id = "gpt2"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 # model = AutoModelForCausalLM.from_pretrained(model_id)
@@ -44,6 +45,7 @@ pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tok
 hf = HuggingFacePipeline(pipeline=pipe)
 
 
+print( "initalized second model")
 
 # tokenizer = AutoTokenizer.from_pretrained(model_id)
 tokenizer.use_default_system_prompt = False