Spaces:

jonathanjordan21
/

chat_with_me

Running

jonathanjordan21 commited on Jun 17

Commit

000d53c

•

1 Parent(s): 4a7b77c

Update custom_llm.py

Files changed (1) hide show

custom_llm.py CHANGED Viewed

@@ -115,6 +115,7 @@ class CustomLLM(LLM):
         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

         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