doubledsbv
commited on
Commit
•
fed235b
1
Parent(s):
2e2bec1
Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ import transformers
|
|
5 |
import torch
|
6 |
from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
|
7 |
|
8 |
-
model_id = "doubledsbv/Llama-3-Kafka-8B-v0.
|
9 |
|
10 |
model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16)
|
11 |
tokenizer = transformers.AutoTokenizer.from_pretrained(model_id)
|
@@ -56,7 +56,7 @@ gr.ChatInterface(
|
|
56 |
chat_function,
|
57 |
chatbot=gr.Chatbot(height=500),
|
58 |
textbox=gr.Textbox(placeholder="Enter message here", container=False, scale=5),
|
59 |
-
title="Llama-3-Kafka-8B-v0.
|
60 |
description="""
|
61 |
German-focused finetuned version of Llama-3-8B
|
62 |
""",
|
|
|
5 |
import torch
|
6 |
from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
|
7 |
|
8 |
+
model_id = "doubledsbv/Llama-3-Kafka-8B-v0.1"
|
9 |
|
10 |
model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16)
|
11 |
tokenizer = transformers.AutoTokenizer.from_pretrained(model_id)
|
|
|
56 |
chat_function,
|
57 |
chatbot=gr.Chatbot(height=500),
|
58 |
textbox=gr.Textbox(placeholder="Enter message here", container=False, scale=5),
|
59 |
+
title="Llama-3-Kafka-8B-v0.1",
|
60 |
description="""
|
61 |
German-focused finetuned version of Llama-3-8B
|
62 |
""",
|