bugfix to Error: "str" object has no attribute "pad_token_id"

app.py

@@ -7,6 +7,7 @@ from datetime import datetime
 import gradio as gr
 import torch
 from transformers import pipeline, TextIteratorStreamer
+from transformers import AutoTokenizer
 from duckduckgo_search import DDGS
 import spaces  # Import spaces early to enable ZeroGPU support
 
@@ -55,12 +56,13 @@ def load_pipeline(model_name):
     if model_name in PIPELINES:
         return PIPELINES[model_name]
     repo = MODELS[model_name]["repo_id"]
+    tokenizer = AutoTokenizer.from_pretrained(repo)
     for dtype in (torch.bfloat16, torch.float16, torch.float32):
         try:
             pipe = pipeline(
                 task="text-generation",
                 model=repo,
-                tokenizer=repo,
+                tokenizer=tokenizer,
                 trust_remote_code=True,
                 torch_dtype=dtype,
                 device_map="auto"
@@ -73,7 +75,7 @@ def load_pipeline(model_name):
     pipe = pipeline(
         task="text-generation",
         model=repo,
-        tokenizer=repo,
+        tokenizer=tokenizer,
         trust_remote_code=True,
         device_map="auto"
     )