antonovmaxim commited on
Commit
d843ea2
·
1 Parent(s): 8c93272

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -9
app.py CHANGED
@@ -1,10 +1,9 @@
1
  import gradio as gr
2
- from transformers import pipeline
3
-
4
- generator = pipeline('text-generation', model='gpt2')
5
-
6
- def generate_text(prompt):
7
- return generator(prompt, max_length=50)['generated_text']
8
-
9
- iface = gr.Interface(fn=generate_text, inputs="text", outputs="text", title="GPT-2 Text Generation")
10
- iface.launch()
 
1
  import gradio as gr
2
+ def classify():
3
+ return 1
4
+ def img_classify(input_img):
5
+ s = "Вероятность того, что изображение сгенерировано нейросетью равна: " + str(classify(input_img))
6
+ return s
7
+ output1 = gr.inputs.Textbox(placeholder="Результат")
8
+ gui = gr.Interface(fn = img_classify, inputs="image", outputs=output1)
9
+ gui.launch()