OpeoluwaAdekoya commited on
Commit
b45f76a
1 Parent(s): 11b9c39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import gradio as gr
2
- from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
3
  import torch
4
  # Define the repository where your model is saved
5
  model_repo = "Dumele/viv-updated" # Replace with your actual repository
 
1
  import gradio as gr
2
+ from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline, GPTQConfig
3
  import torch
4
  # Define the repository where your model is saved
5
  model_repo = "Dumele/viv-updated" # Replace with your actual repository