Update app.py

app.py

@@ -16,7 +16,9 @@ model_name = "meta-llama/Llama-3.2-1B"
 model = AutoModelForCausalLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
-pipe = pipeline(task = 'text-generation', model=model, tokenizer=tokenizer, max_length = 512) # Check documentation without "feature-extraction"
+pipe = pipeline(model=model, tokenizer=tokenizer, max_length = 512) # Check documentation without "feature-extraction"
+#pipe = pipeline(task = 'text-generation', model=model, tokenizer=tokenizer, max_length = 512) # Check documentation without "feature-extraction"
+
 
 # Use transformers pipeline
 llm_pipeline = HuggingFacePipeline(pipeline=pipe)