Uncover-F commited on
Commit
741290e
·
verified ·
1 Parent(s): f65b360

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@ import torch
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
3
  import gradio as gr
4
 
5
- model_id = "AlphaMindQ/GPTx-v6-60gb"
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 GPTx-v6-60GB",
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()