Fix return values in respond function and add photo

app.py

@@ -19,6 +19,10 @@ model_name = "dicta-il/dictalm2.0-instruct"
 model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
+# Set the pad_token to eos_token if not already set
+if tokenizer.pad_token is None:
+    tokenizer.pad_token = tokenizer.eos_token
+
 @app.route('/chat', methods=['POST'])
 def chat():
     data = request.json
@@ -33,7 +37,7 @@ def chat():
     inputs = tokenizer(user_input, return_tensors='pt', padding=True, truncation=True)
     input_ids = inputs['input_ids']
     attention_mask = inputs['attention_mask']
-    outputs = model.generate(input_ids, attention_mask=attention_mask, pad_token_id=tokenizer.eos_token_id)
+    outputs = model.generate(input_ids, attention_mask=attention_mask)
     response_text = tokenizer.decode(outputs[0], skip_special_tokens=True).replace(user_input, '').strip()
     
     return jsonify({"response": response_text})