Shreyas094 commited on
Commit
5f39768
·
verified ·
1 Parent(s): 8994bce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -51,8 +51,8 @@ class HuggingFaceHubProvider(LlamaCppEndpointSettings):
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
 
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