fix

app.py

@@ -19,9 +19,7 @@ from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer, set_seed
 
 
 def inference(input_sentence, max_length, sample_or_greedy, seed=42):
-
-    print("input_sentence", input_sentence)
-
+    #print("input_sentence", input_sentence)
     if sample_or_greedy == "Sample":
         parameters = {
             "max_new_tokens": max_length,
@@ -63,7 +61,7 @@ def inference(input_sentence, max_length, sample_or_greedy, seed=42):
     #generation = data[0]["generated_text"].split(input_sentence, 1)[1]
 
     generation  = res[0]["generated_text"].split(input_sentence, 1)[1]
-    #print(generation)
+    print(generation)
 
     return (
         before_prompt