Update app.py

app.py

@@ -66,7 +66,7 @@ def RunLegalModel(model, text):
     imgclassifier  = pettyfogger("ext-generation", model=model)
     if image is not None:  
         description = pettyfogger(text)
-    return description
+    return description[0]['generated_text']
     
 radio1 = gr.Radio(["microsoft/resnet-50", "google/vit-base-patch16-224", "apple/mobilevit-small"], value="microsoft/resnet-50", label="Select a Classifier", info="Image Classifier")
 tab1 = gr.Interface(
@@ -108,15 +108,15 @@ tab4 = gr.Interface(
 )
 
 
-classifiertypes = ["umarbutler/open-australian-legal-distilgpt2"]
-radio5 = gr.Radio(classifiertypes, value="umarbutler/open-australian-legal-distilgpt2", label="Select", info="Legal Question")
+classifiertypes = ["umarbutler/open-australian-legal-llm"]
+radio5 = gr.Radio(classifiertypes, value="umarbutler/open-australian-legal-llm", label="Select", info="Legal Question")
 textinput5 = gr.Textbox(value="Under the purposes of Part 6 Division 2 of the Act, regulations may confer power on an applicant for")
 
 
 tab5 = gr.Interface(
     fn=RunLegalModel,
     inputs=[radio5, textinput5],
-    outputs=[text],
+    outputs=["text"],
 )
 
 demo = gr.TabbedInterface([tab1, tab2, tab3, tab4, tab5], ["Describe", "Estimage Age", "Speak", "Generate Image", "Aus. Legal"])