Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -2,7 +2,7 @@ import torch
|
|
| 2 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 3 |
import gradio as gr
|
| 4 |
|
| 5 |
-
model_id = "
|
| 6 |
|
| 7 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 8 |
model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float32)
|
|
@@ -34,6 +34,6 @@ gr.Interface(
|
|
| 34 |
gr.Slider(0.5, 2.0, value=1.0, step=0.1, label="Repetition Penalty")
|
| 35 |
],
|
| 36 |
outputs=gr.Textbox(label="Generated Text"),
|
| 37 |
-
title="🧠 AlphaMindQ
|
| 38 |
theme="default"
|
| 39 |
).launch()
|
|
|
|
| 2 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 3 |
import gradio as gr
|
| 4 |
|
| 5 |
+
model_id = "distilbert/distilgpt2"
|
| 6 |
|
| 7 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 8 |
model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float32)
|
|
|
|
| 34 |
gr.Slider(0.5, 2.0, value=1.0, step=0.1, label="Repetition Penalty")
|
| 35 |
],
|
| 36 |
outputs=gr.Textbox(label="Generated Text"),
|
| 37 |
+
title="🧠 AlphaMindQ Fork distilbert/distilgpt2",
|
| 38 |
theme="default"
|
| 39 |
).launch()
|