Update app.py

app.py

@@ -8,21 +8,23 @@ model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 generator = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
 
 def generate_json(prompt):
-    instruction = f"Generate a JSON object with the following properties: {prompt}"
+    # Make the instruction explicit: return ONLY JSON, without explanation
+    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}"
+    )
     result = generator(instruction, max_length=256, do_sample=False)
-    generated_text = result[0]["generated_text"]
+    generated_text = result[0]["generated_text"].strip()  # remove extra whitespace
+    
+    print(f"Raw Model Output: {generated_text}")  # Debug statement
     
-    # Debug: Print raw model output to inspect
-    print(f"Raw Model Output: {generated_text}")
-
     try:
-        # Try parsing the generated text as JSON
         parsed = json.loads(generated_text)
         formatted_json = json.dumps(parsed, indent=2)
     except Exception as e:
-        # If parsing fails, print the error
         formatted_json = f"Raw Output:\n{generated_text}\n\nError parsing JSON: {e}"
-
+    
     return formatted_json
 
 demo = gr.Interface(