Update app.py

app.py

@@ -3,7 +3,7 @@ from unsloth import FastLanguageModel
 import torch
 
 # Load the model and tokenizer
-model_name = "Rafay17/Llama3.2_1b_customModle2"  # Use your specific model name
+model_name = "Rafay17/Llama3.2_1b_customModle2"  # Your model name
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 model = FastLanguageModel.from_pretrained(
@@ -18,38 +18,7 @@ FastLanguageModel.for_inference(model)  # Call this immediately after loading th
 # Function to generate a response
 def generate_response(input_text):
     # Prepare the labeled prompt for the model
-    labeled_prompt = (
-        "Please provide the response with the following labels:\n"
-        "Speaker: [SPEAKER]\n"
-        "Text: [TEXT]\n"
-        "Sentiment: [SENTIMENT]\n"
-        "Emotion: [EMOTION]\n"
-        "Intent: [INTENT]\n"
-        "Tone: [TONE]\n"
-        "Confidence Level: [CONFIDENCE]\n"
-        "Frustration Level: [FRUSTRATION]\n"
-        "Response Length: [LENGTH]\n"
-        "Action Required: [ACTION]\n"
-        "Interruption: [INTERRUPTION]\n"
-        "Cooperation Level: [COOPERATION]\n"
-        "Clarity: [CLARITY]\n"
-        "Objective: [OBJECTIVE]\n"
-        "Timeline: [TIMELINE]\n"
-        "Motivation: [MOTIVATION]\n"
-        "Conversation Stage: [STAGE]\n"
-        "Resolution: [RESOLUTION]\n"
-        "Context: [CONTEXT]\n"
-        "Urgency: [URGENCY]\n"
-        "Problem Type: [PROBLEM]\n"
-        "Key Words: [KEYWORDS]\n"
-        "Expected Detail: [DETAIL]\n"
-        "Time Gap: [TIME]\n"
-        "Client Expectation: [EXPECTATION]\n"
-        "Channel: [CHANNEL]\n"
-        "Power Relationship: [POWER]\n\n"
-        f"User Input: {input_text}\n"
-        "Response:"
-    )
+    labeled_prompt = f"User Input: {input_text}\nResponse:"
 
     # Prepare the input for the model
     inputs = tokenizer(
@@ -75,12 +44,13 @@ def generate_response(input_text):
 
 # Function to take user input and generate output
 def user_interaction():
+    print("Welcome to the Chatbot! Type 'exit' to quit.")
     while True:
-        user_input = input("Enter conversation details (or type 'exit' to quit): ")
+        user_input = input("You: ")
         if user_input.lower() == 'exit':
-            print("Exiting the program.")
+            print("Exiting the chatbot. Goodbye!")
             break
-        print("Generating response for input:")
+        print("Chatbot is generating a response...")
         generate_response(user_input)
 
 # Start the user interaction