redael commited on
Commit
b256eab
·
verified ·
1 Parent(s): f7537f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,8 +1,8 @@
1
 
 
2
  import torch
3
  import gradio as gr
4
 
5
- # Replace 'your-username/your-model-name' with the actual model name you uploaded to Hugging Face
6
  model_name = 'redael/model_udc'
7
 
8
  # Load the tokenizer and model from Hugging Face
@@ -40,3 +40,4 @@ iface = gr.Interface(fn=chatbot_interface, inputs="text", outputs="text", title=
40
 
41
  # Launch the Gradio interface
42
  iface.launch()
 
 
1
 
2
+ from transformers import AutoModelForCausalLM, AutoTokenizer
3
  import torch
4
  import gradio as gr
5
 
 
6
  model_name = 'redael/model_udc'
7
 
8
  # Load the tokenizer and model from Hugging Face
 
40
 
41
  # Launch the Gradio interface
42
  iface.launch()
43
+