Spaces:
Runtime error
Runtime error
Commit
·
35e895f
1
Parent(s):
134c118
Update components/custom_llm.py
Browse files- 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']
|