Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 
-import os
+import os, sys
 from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig, pipeline
 import torch
 
@@ -49,9 +49,9 @@ def respond(
         for hmessage in messages:
             role = "<|assistant|>" if hmessage['role'] == 'assistant' else "<|user|>"
             prompt += f"\n{role}\n{hmessage['content']}\n<|end|>\n"        
-        prompt += f"\n<|user|>\n{message}\n<|end|><|assistant|>\n"
+        # prompt += f"\n<|user|>\n{message}\n<|end|><|assistant|>\n"
 
-        """
+        # """
         # Generate the response
         response_output = generator(
             prompt,
@@ -68,10 +68,12 @@ def respond(
     
         # Extract the assistant's response
         result = generated_text[len(prompt):].strip()
-        """
-        result = message+':'+prompt
+        # """
+        # result = message+':'+prompt
     except Exception as error:
-        result = str(error)
+        exc_type, exc_obj, exc_tb = sys.exc_info()
+        fname = os.path.split(exc_tb.tb_frame.f_code.co_filename)[1]
+        result = str(error) +':'+ exc_type +':'+ fname +':'+ exc_tb.tb_lineno
 
     yield result