Update app.py

app.py

@@ -61,7 +61,13 @@ def ImageGenFromText(text, model):
         print(image)
         return image
 
-
+@spaces.GPU
+def RunLegalModel(model, text):
+    imgclassifier  = pettyfogger("ext-generation", model=model)
+    if image is not None:  
+        description = pettyfogger(text)
+    return description
+    
 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(
     fn=guessanImage,
@@ -104,7 +110,7 @@ tab4 = gr.Interface(
 
 classifiertypes = ["umarbutler/open-australian-legal-distilgpt2"]
 radio5 = gr.Radio(classifiertypes, value="umarbutler/open-australian-legal-distilgpt2", label="Select", info="Legal Question")
-textinput5 = r.Textbox(value="Under the purposes of Part 6 Division 2 of the Act, regulations may confer power on an applicant for")
+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(