aal-hawa commited on
Commit
d1910dc
·
1 Parent(s): 1d9ee70
Files changed (1) hide show
  1. app.py +15 -1
app.py CHANGED
@@ -27,10 +27,24 @@ def load_model():
27
  model.eval()
28
  print("Model loaded.")
29
 
 
 
 
 
 
 
 
 
 
 
 
 
30
  def chat_response(message, history):
31
  load_model()
32
 
33
- messages = history + [{"role": "user", "content": message}]
 
 
34
 
35
  text = tokenizer.apply_chat_template(
36
  messages,
 
27
  model.eval()
28
  print("Model loaded.")
29
 
30
+ def normalize_content(msg):
31
+ """Convert list content to string (Gradio may pass content as a list)."""
32
+ if isinstance(msg.get("content"), list):
33
+ parts = []
34
+ for item in msg["content"]:
35
+ if isinstance(item, dict) and "text" in item:
36
+ parts.append(item["text"])
37
+ elif isinstance(item, str):
38
+ parts.append(item)
39
+ return {"role": msg["role"], "content": " ".join(parts)}
40
+ return msg
41
+
42
  def chat_response(message, history):
43
  load_model()
44
 
45
+ # Normalize history: convert any list content to plain strings
46
+ clean_history = [normalize_content(m) for m in history]
47
+ messages = clean_history + [{"role": "user", "content": message}]
48
 
49
  text = tokenizer.apply_chat_template(
50
  messages,