Fixed the output

app.py

@@ -36,12 +36,14 @@ def ocr(input_img):
     inputs = processor(image, prompt, return_tensors="pt").to(model.device)
 
     output = model.generate(**inputs, max_new_tokens=200)
-    return processor.decode(output[0])
+    res = processor.decode(output[0])
+    return res[len(prompt):res.find("<ORC>") - 5]
 
 
-prompt = '''Review the following essay and score it. Output the score and the reasoning.The output format is:
+prompt = '''Review the following essay and score it. The output format is:
 Score: *<score>*
-Reasoning: 1... 2... 3... 4... 5...
+Reason: ...
+Suggestions: ...
 '''
 
 def score_with_gpt(text):
@@ -73,7 +75,7 @@ def score_it(input_img):
 
 demo = gr.Interface(fn=score_it, title="Upload your English script and get the score",
                     inputs=[gr.Image()],
-                    outputs=[gr.Textbox(label="Text", lines=10), gr.Textbox(
+                    outputs=[gr.Textbox(label="Text", lines=10), gr.Markdown(
                         label="Score", lines=10)],
                     stop_btn="Stop Generation",
                     )