jonathanjordan21 commited on
Commit
35e895f
·
1 Parent(s): 134c118

Update components/custom_llm.py

Browse files
Files changed (1) hide show
  1. components/custom_llm.py +0 -1
components/custom_llm.py CHANGED
@@ -67,5 +67,4 @@ class CustomLLM(LLM):
67
 
68
  data = {"inputs": prompt, "parameters":parameters_dict, "options":{"wait_for_model":True}}
69
  data = requests.post(API_URL, headers=headers, json=data).json()
70
- print(data)
71
  return data[0]['generated_text']
 
67
 
68
  data = {"inputs": prompt, "parameters":parameters_dict, "options":{"wait_for_model":True}}
69
  data = requests.post(API_URL, headers=headers, json=data).json()
 
70
  return data[0]['generated_text']