Update app.py

app.py

@@ -10,7 +10,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, GPTQConfig
 import torch
 
 # Define the repository where your model is saved
-model_repo = "Dumele/autotrain-shhsb-57a2l"  # Replace with your actual repository
+model_repo = "Dumele/viv-updated2"  # Replace with your actual repository
 
 # Load the tokenizer from the repository
 tokenizer = AutoTokenizer.from_pretrained(model_repo)
@@ -49,7 +49,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 import gradio as gr
 
 # Define the repository where your model is saved
-model_repo = "Dumele/autotrain-shhsb-57a2l"  # Replace with your actual repository name
+model_repo = "Dumele/viv-updated2"  # Replace with your actual repository name
 
 # Load the tokenizer from the repository
 tokenizer = AutoTokenizer.from_pretrained(model_repo)