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