Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ import gradio as gr
|
|
5 |
from transformers import GenerationConfig, LlamaForCausalLM, LlamaTokenizer
|
6 |
from transformers import Trainer
|
7 |
|
8 |
-
BASE_MODEL = "TheBloke/
|
9 |
|
10 |
# Create a custom device map
|
11 |
# This will vary based on the architecture of model and the memory capacity of GPU and CPU
|
@@ -54,6 +54,6 @@ iface = gr.Interface(
|
|
54 |
inputs="text",
|
55 |
outputs="text",
|
56 |
title="Chatbot",
|
57 |
-
description="This vicuna app is using this model: https://huggingface.co/TheBloke/
|
58 |
)
|
59 |
iface.launch()
|
|
|
5 |
from transformers import GenerationConfig, LlamaForCausalLM, LlamaTokenizer
|
6 |
from transformers import Trainer
|
7 |
|
8 |
+
BASE_MODEL = "TheBloke/vicuna-7B-1.1-HF"
|
9 |
|
10 |
# Create a custom device map
|
11 |
# This will vary based on the architecture of model and the memory capacity of GPU and CPU
|
|
|
54 |
inputs="text",
|
55 |
outputs="text",
|
56 |
title="Chatbot",
|
57 |
+
description="This vicuna app is using this model: https://huggingface.co/TheBloke/vicuna-7B-1.1-HF"
|
58 |
)
|
59 |
iface.launch()
|