Update app.py

app.py

@@ -32,9 +32,14 @@ tokenizer = AutoTokenizer.from_pretrained(
     token=api_token
 )
 
-# Ensure eos_token_id is defined
+# Ensure eos_token_id is set
 if model.generation_config.eos_token_id is None:
-    model.generation_config.eos_token_id = tokenizer.eos_token_id
+    eos_token_id = tokenizer.eos_token_id or tokenizer.pad_token_id
+    if eos_token_id is None:
+        raise ValueError(
+            "Both `eos_token_id` and `pad_token_id` are undefined in the tokenizer. Please define one explicitly."
+        )
+    model.generation_config.eos_token_id = eos_token_id
 
 # Preprocess image
 def preprocess_image(image):