doubledsbv commited on
Commit
fed235b
1 Parent(s): 2e2bec1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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.2"
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.2",
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
  """,