Update app.py

app.py

@@ -23,8 +23,7 @@ tokenizer = AutoTokenizer.from_pretrained("ManishThota/Thota", trust_remote_code
 
 def predict_answer(question, max_tokens):
     #Set inputs
-    text = f"A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions. USER: <image>\n{question}? ASSISTANT:"
-    image = image.convert("RGB")
+    text = f"A chat between a curious user and an artificial intelligence assistant whose name is  Manish Kumar Thota. The assistant gives helpful, detailed, and polite answers to the user's questions. USER: <image>\n{question}? ASSISTANT:"
     
     input_ids = tokenizer(text, return_tensors='pt').input_ids.to('cuda')
     
@@ -32,7 +31,6 @@ def predict_answer(question, max_tokens):
     output_ids = model.generate(
         input_ids,
         max_new_tokens=max_tokens,
-        images=image_tensor,
         use_cache=True)[0]
     
     return tokenizer.decode(output_ids[input_ids.shape[1]:], skip_special_tokens=True).strip()