add interface

.ipynb_checkpoints/app-checkpoint.py

@@ -1,7 +1,21 @@
+import os
+API_TOKEN = os.getenv('API_TOKEN')
+
 import gradio as gr
+import requests
+
+API_URL = "https://api-inference.huggingface.co/models/tlkh/flan-t5-paraphrase-classify-explain"
+headers = {"Authorization": f"Bearer {API_TOKEN}"}
+
+def query(payload):
+    response = requests.post(API_URL, headers=headers, json=payload)
+    return response.json()
+
+def infer(s1, s2):
+    model_input = "Classify and explain the relationship between this pair of sentences: <S1> "+s1+" </S1><S2> "+s2+" </S2>"
+    data = query(payload)
+    return data
 
-def greet(name):
-    return "Hello " + name + "!!"
+demo = gr.Interface(fn=infer, inputs=["text", "text"], outputs="text")
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
 demo.launch()

.ipynb_checkpoints/requirements-checkpoint.txt

@@ -1,2 +1,3 @@
 gradio
 transformers
+requests

app.py

@@ -1,7 +1,21 @@
+import os
+API_TOKEN = os.getenv('API_TOKEN')
+
 import gradio as gr
+import requests
+
+API_URL = "https://api-inference.huggingface.co/models/tlkh/flan-t5-paraphrase-classify-explain"
+headers = {"Authorization": f"Bearer {API_TOKEN}"}
+
+def query(payload):
+    response = requests.post(API_URL, headers=headers, json=payload)
+    return response.json()
+
+def infer(s1, s2):
+    model_input = "Classify and explain the relationship between this pair of sentences: <S1> "+s1+" </S1><S2> "+s2+" </S2>"
+    data = query(payload)
+    return data
 
-def greet(name):
-    return "Hello " + name + "!!"
+demo = gr.Interface(fn=infer, inputs=["text", "text"], outputs="text")
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
 demo.launch()

requirements.txt

@@ -1,2 +1,3 @@
 gradio
 transformers
+requests