Update app.py

app.py

@@ -1,15 +1,9 @@
 from fastapi import FastAPI
 # Use a pipeline as a high-level helper
-#from transformers import pipeline
-#from huggingface_hub import login
-from transformers import AutoTokenizer, AutoModelForCausalLM
+from transformers import pipeline
 
-tokenizer = AutoTokenizer.from_pretrained("mistralai/Mixtral-8x7B-v0.1")
-model = AutoModelForCausalLM.from_pretrained("mistralai/Mixtral-8x7B-v0.1")
-login()
-
-#pipe = pipeline("text-generation", model="meta-llama/Llama-3.2-3B")
 
+pipe = pipeline("text-generation", model="deepseek-ai/DeepSeek-V2.5-1210", trust_remote_code=True)
 app = FastAPI()