Update app.py

app.py

@@ -41,7 +41,14 @@ def generate_text(prompt, temperature, top_p):
         generated_text += decoded_token  # Append the new token to the generated text
         if decoded_token == "<":  # Stop if the end of sequence token is generated
             break
-        yield generated_text[prompt_length:]  # Yield the generated text excluding the initial prompt plus "EOS"
+    
+    # Check if the generated text contains "0" or "1" and return the appropriate sentiment message
+    if "1" in generated_text:
+        return "The sentiment is positive."
+    elif "0" in generated_text:
+        return "The sentiment is negative."
+    else:
+        return "Invalid"
 
 # Create a Gradio interface with a text input, sliders for temperature and top_p, and a text output
 interface = gr.Interface(
@@ -56,4 +63,4 @@ interface = gr.Interface(
     description="SentimentGPT processes the sequence and returns a reasonably accurate guess of whether the sentiment behind the input is positive or negative."
 )
 
-interface.launch()
+interface.launch()