Jordan Bracco commited on
Commit
c266f8b
1 Parent(s): bc58bd1

no idea what i'm doing, but doing it!

Browse files
Files changed (2) hide show
  1. app.py +41 -0
  2. requirements.txt +3 -0
app.py ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ import gradio as gr
3
+ import requests
4
+ import os
5
+ import transformers
6
+ from transformers import AutoModelWithLMHead, AutoTokenizer, AutoModelForCausalLM, pipeline
7
+ from transformers import GPT2Tokenizer, GPT2Model
8
+
9
+ tokenizer = AutoTokenizer.from_pretrained("href/gpt2-schiappa")
10
+
11
+ model = AutoModelForCausalLM.from_pretrained("href/gpt2-schiappa")
12
+ pipe = pipeline('text-generation', model="href/gpt2-schiappa", tokenizer=tokenizer)
13
+
14
+ def text_generation(input_text, seed, min_length, max_length, temperature, top_k, top_p, repetition_penalty):
15
+ input_ids = tokenizer(input_text, return_tensors="pt").input_ids
16
+ torch.manual_seed(seed) # Max value: 18446744073709551615
17
+ outputs = model.generate(input_ids, do_sample=True, max_length=max_length, min_length=min_length, temperature=temperature, top_k=top_k, top_p=top_p, repetition_penalty=repetition_penalty)
18
+ generated_text = tokenizer.batch_decode(outputs, skip_special_tokens=True)
19
+ return generated_text
20
+
21
+ title = "Minelli"
22
+ description = "Écrit comme Marie Minelli"
23
+
24
+ gr.Interface(
25
+ text_generation,
26
+ [
27
+ gr.inputs.Textbox(lines=2, label="Prompt"),
28
+ gr.inputs.Number(default=10, label="Enter seed number"),
29
+ gr.inputs.Slider(default=0, label="Minimum Length", minimum=0),
30
+ gr.inputs.Slider(default=100, label="Maximum Length", minimum=0),
31
+ gr.inputs.Slider(default=1.0, label="Temperature", minimum=0.1),
32
+ gr.inputs.Slider(default=40, label="topK", minimum=1),
33
+ gr.inputs.Slider(default=0.9, label="topP", minimum=0.0, maximum=1.0),
34
+ gr.inputs.Slider(default=1.0, label="Repetition penalty", minimum=1.0)
35
+ ],
36
+ [gr.outputs.Textbox(type="auto", label="Generated")],
37
+ title=title,
38
+ description=description,
39
+ theme="huggingface"
40
+ ).launch()
41
+
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ torch
2
+ transformers
3
+ gradio