Spaces:

alexkueck
/

TestInferenceAPI

Sleeping

alexkueck commited on Dec 19, 2023

Commit

55c9502

1 Parent(s): fcac988

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -278,7 +278,7 @@ def generate(text, history, rag_option, model_option,  k=3, temperature=0.5, max
             API_URL = "https://api-inference.huggingface.co/models/HuggingFaceH4/zephyr-7b-beta"
             print("HF1")
         else:
-            API_URL = "https://api-inference.huggingface.co/models/tiiuae/falcon-7b-instruct"
             print("HF2")
         if (rag_option == "An"):

             API_URL = "https://api-inference.huggingface.co/models/HuggingFaceH4/zephyr-7b-beta"
             print("HF1")
         else:
+            API_URL = "https://api-inference.huggingface.co/models/tiiuae/falcon-180B-chat"
             print("HF2")
         if (rag_option == "An"):