Spaces:
Sleeping
Sleeping
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
|