Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -51,8 +51,8 @@ class HuggingFaceHubProvider(LlamaCppEndpointSettings):
|
|
| 51 |
self.model = model
|
| 52 |
|
| 53 |
def create_completion(self, prompt, **kwargs):
|
| 54 |
-
|
| 55 |
-
|
| 56 |
|
| 57 |
def get_provider_default_settings(self):
|
| 58 |
return self.model.model_kwargs
|
|
|
|
| 51 |
self.model = model
|
| 52 |
|
| 53 |
def create_completion(self, prompt, **kwargs):
|
| 54 |
+
response = self.model(prompt)
|
| 55 |
+
return {'choices': [{'text': response['generated_text']}]}
|
| 56 |
|
| 57 |
def get_provider_default_settings(self):
|
| 58 |
return self.model.model_kwargs
|