Update app.py

app.py

@@ -57,7 +57,7 @@ from langchain.llms import HuggingFacePipeline
 import torch
 
 # model_name = "deepseek-ai/deepseek-llm-7b-chat"
-model_name = "google/flan-t5-small"
+model_name = "mistralai/tiny-random-Mistral-7B"
 
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 
@@ -70,7 +70,7 @@ model = AutoModelForCausalLM.from_pretrained(
 )
 
 hf_pipeline = pipeline(
-    "text2text-generation",
+    "text-generation",
     model=model,
     tokenizer=tokenizer,
     max_new_tokens=300