Yhhxhfh commited on
Commit
f011203
1 Parent(s): 5097ad2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -48,7 +48,7 @@ class ModelManager:
48
  response = requests.get(download_url, headers={"Authorization": f"Bearer {HUGGINGFACE_TOKEN}"})
49
  with open(model_path, 'wb') as f:
50
  f.write(response.content)
51
- model = Llama.from_pretrained(model_path, **self.params)
52
  models.append(model)
53
 
54
  self.params["tokens"] = models[0].tokenize(b"Hello")
@@ -99,4 +99,4 @@ iface = gr.Interface(
99
  )
100
 
101
  if __name__ == "__main__":
102
- iface.launch()
 
48
  response = requests.get(download_url, headers={"Authorization": f"Bearer {HUGGINGFACE_TOKEN}"})
49
  with open(model_path, 'wb') as f:
50
  f.write(response.content)
51
+ model = Llama(model_path=model_path, **self.params)
52
  models.append(model)
53
 
54
  self.params["tokens"] = models[0].tokenize(b"Hello")
 
99
  )
100
 
101
  if __name__ == "__main__":
102
+ iface.launch()