Spaces:

dixiyao
/

chatbot-professor

Sleeping

dixiyao commited on Jul 9

Commit

378d4ec

•

1 Parent(s): c617dc5

no toxic

Files changed (1) hide show

rag.py CHANGED Viewed

@@ -33,7 +33,7 @@ class Llama3_8B_gen:
     def __init__(self, threshold):
         self.pipeline = transformers.pipeline(
             "text-generation",
-            model="dixiyao/toxic-professor-bot-dpo_3epochs",
             token=os.getenv("HF_TOKEN"),
             model_kwargs={
                 "torch_dtype": torch.float16,

     def __init__(self, threshold):
         self.pipeline = transformers.pipeline(
             "text-generation",
+            model="dixiyao/professor-bot-dpo_3epochs",
             token=os.getenv("HF_TOKEN"),
             model_kwargs={
                 "torch_dtype": torch.float16,