Update app.py
Browse files
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 |
+
|