Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|
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()
|