Update app.py

app.py

@@ -7,11 +7,14 @@ from huggingface_hub import InferenceClient
 #from transformers import pipeline
 import torch
 
-# Load model directly
-from transformers import AutoTokenizer, AutoModelForCausalLM
+# Use a pipeline as a high-level helper
+from transformers import pipeline
 
-tokenizer = AutoTokenizer.from_pretrained("deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B")
-model = AutoModelForCausalLM.from_pretrained("deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B")
+messages = [
+    {"role": "user", "content": "Who are you?"},
+]
+pipe = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B")
+pipe(messages)
 
 def chat(input_text, history=[]):
     history.append(input_text)