Update app.py

app.py

@@ -4,14 +4,14 @@
 import base64
 import gradio as gr
 from huggingface_hub import InferenceClient
-from transformers import pipeline
+#from transformers import pipeline
 import torch
 
-messages = [
-    {"role": "user", "content": "Who are you?"},
-]
-pipe = pipeline("image-text-to-text", model="Qwen/Qwen2.5-VL-3B-Instruct")
-pipe(messages)
+# Load model directly
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+tokenizer = AutoTokenizer.from_pretrained("deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B")
+model = AutoModelForCausalLM.from_pretrained("deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B")
 
 def chat(input_text, history=[]):
     history.append(input_text)