Upload app.py

app.py

@@ -4,10 +4,17 @@ from peft import AutoPeftModelForSequenceClassification
 
 tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
 loraModel = AutoPeftModelForSequenceClassification.from_pretrained("Intradiction/text_classification_WithLORA")
+
+#pretrained models
+#Textclass_pipe = pipeline()
+#STSmodel_pipe = pipeline()
+#NLImodel_pipe = pipeline()
+
 # Handle calls to DistilBERT no LORA
 distilBERTnoLORA_pipe = pipeline(model="Intradiction/text_classification_NoLORA")
 distilBERTwithLORA_pipe = pipeline("sentiment-analysis", model=loraModel, tokenizer=tokenizer)
 
+
 def distilBERTnoLORA_fn(text):
     return distilBERTnoLORA_pipe(text)
 
@@ -106,15 +113,25 @@ with gr.Blocks(
                             <p>insert information on training parameters here</p>
                             """)
             with gr.Column(scale=0.3,variant="panel"):
-                inp = gr.Textbox(placeholder="Prompt",label= "Enter Query")
+                nli_p1 = gr.Textbox(placeholder="Prompt One",label= "Enter Query")
+                nli_p2 = gr.Textbox(placeholder="Prompt Two",label= "Enter Query")
                 btn = gr.Button("Run")
                 gr.Examples(
                     [
-                        "I thought this was a bit contrived",
-                        "You would need to be a child to enjoy this",
-                        "Drive more like Drive away",
+                        "placeholder text",
+                        "placeholder text",
+                        "placeholder text",
                     ],
-                    inp,
+                    nli_p1,
+                    label="Try asking",
+                ) 
+                gr.Examples(
+                    [
+                        "placeholder text",
+                        "placeholder text",
+                        "placeholder text",
+                    ],
+                    nli_p2,
                     label="Try asking",
                 ) 
 
@@ -153,15 +170,25 @@ with gr.Blocks(
                             <p>insert information on training parameters here</p>
                             """)
             with gr.Column(scale=0.3,variant="panel"):
-                inp = gr.Textbox(placeholder="Prompt",label= "Enter Query")
+                sts_p1 = gr.Textbox(placeholder="Prompt One",label= "Enter Query")
+                sts_p2 = gr.Textbox(placeholder="Prompt Two",label= "Enter Query")
                 btn = gr.Button("Run")
                 gr.Examples(
                     [
-                        "I thought this was a bit contrived",
-                        "You would need to be a child to enjoy this",
-                        "Drive more like Drive away",
+                        "placeholder text",
+                        "placeholder text",
+                        "placeholder text",
                     ],
-                    inp,
+                    sts_p1,
+                    label="Try asking",
+                )
+                gr.Examples(
+                    [
+                        "placeholder text",
+                        "placeholder text",
+                        "placeholder text",
+                    ],
+                    sts_p2,
                     label="Try asking",
                 )