Spaces:

jonathanjordan21
/

chat_with_me

Running

jonathanjordan21 commited on Jan 6, 2024

Commit

989adae

1 Parent(s): 0c56ceb

Update custom_llm.py

Files changed (1) hide show

custom_llm.py CHANGED Viewed

@@ -112,7 +112,6 @@ class CustomLLM(LLM):
         data = {"inputs": prompt, "parameters":parameters_dict, "options":{"wait_for_model":True}}
         data = requests.post(API_URL, headers=headers, json=data).json()
-        print(data)
         return data[0]['generated_text']
     @property

         data = {"inputs": prompt, "parameters":parameters_dict, "options":{"wait_for_model":True}}
         data = requests.post(API_URL, headers=headers, json=data).json()
         return data[0]['generated_text']
     @property