Spaces:
Runtime error
Runtime error
Commit
·
989adae
1
Parent(s):
0c56ceb
Update custom_llm.py
Browse files- custom_llm.py +0 -1
custom_llm.py
CHANGED
|
@@ -112,7 +112,6 @@ class CustomLLM(LLM):
|
|
| 112 |
|
| 113 |
data = {"inputs": prompt, "parameters":parameters_dict, "options":{"wait_for_model":True}}
|
| 114 |
data = requests.post(API_URL, headers=headers, json=data).json()
|
| 115 |
-
print(data)
|
| 116 |
return data[0]['generated_text']
|
| 117 |
|
| 118 |
@property
|
|
|
|
| 112 |
|
| 113 |
data = {"inputs": prompt, "parameters":parameters_dict, "options":{"wait_for_model":True}}
|
| 114 |
data = requests.post(API_URL, headers=headers, json=data).json()
|
|
|
|
| 115 |
return data[0]['generated_text']
|
| 116 |
|
| 117 |
@property
|