Update app.py

app.py

@@ -8,16 +8,19 @@ model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 generator = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
 
 def generate_json(prompt):
-    # Make the instruction explicit: return ONLY JSON, without explanation
+    # Revise the instruction to make it explicit for valid JSON output
     instruction = (
-        f"Generate only a valid JSON object without any markdown or additional text. "
-        f"The JSON object should have the following keys: title, author, and tags. "
-        f"Fill in dummy values. Now create a JSON object for: {prompt}"
+        "Return only a valid JSON object, with no additional text. "
+        "The JSON object must contain exactly the keys: "
+        '"title", "author", "tags". '
+        "For the following prompt, generate this JSON object. "
+        f"Prompt: {prompt}"
     )
-    result = generator(instruction, max_length=256, do_sample=False)
-    generated_text = result[0]["generated_text"].strip()  # remove extra whitespace
+    result = generator(instruction, max_length=512, do_sample=False)
+    generated_text = result[0]["generated_text"].strip()
     
-    print(f"Raw Model Output: {generated_text}")  # Debug statement
+    # Debug: print the raw output to inspect the format
+    print(f"Raw Model Output: {generated_text}")
     
     try:
         parsed = json.loads(generated_text)