Update app.py

app.py

@@ -4,8 +4,8 @@ import gradio as gr
 import torch
 
 if torch.cuda.is_available():
-    tokenizer = AutoTokenizer.from_pretrained("ai-forever/mGPT-13B")
-    model = AutoModelForCausalLM.from_pretrained("ai-forever/mGPT-13B", load_in_8bit=True, device_map="auto")
+    tokenizer = AutoTokenizer.from_pretrained("ai-forever/ruGPT-3.5-13B")
+    model = AutoModelForCausalLM.from_pretrained("ai-forever/ruGPT-3.5-13B", load_in_8bit=True, device_map="auto")
     pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
 @spaces.GPU
@@ -14,7 +14,7 @@ def predict(prompt, temperature, max_length):
 
 demo = gr.Interface(
     fn=predict,
-    title="mGPT-13B Demo",
+    title="ruGPT-3.5-13B Demo",
     inputs=["text", gr.Slider(minimum=0.01, maximum=1.0, value=0.7, label="temperature"), gr.Slider(minimum=1, maximum=1024, value=50, label="max_length")],
     outputs=["text"],
 )