Add Collinear SDK

app.py

@@ -1,15 +1,5 @@
 import gradio as gr
-# from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
-# from jinja2 import Template
-# import torch
-# import json
-
-# # load the judge
-# device = "cuda:0"
-# model_name = "collinear-ai/collinear-reliability-judge-v8-sauber"
-# model_pipeline = pipeline(task="text-classification", model=model_name, device=device)
-
-# templates
+from collinear import Collinear
 conv_template = Template(
     """
     # Context:

requirements.txt

@@ -2,4 +2,5 @@ jinja2
 sentencepiece
 transformers
 gradio
-torch
+torch
+collinear