Loewolf commited on
Commit
005a724
1 Parent(s): a93d19f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -4
app.py CHANGED
@@ -1,13 +1,19 @@
1
  import gradio as gr
 
2
 
3
- # Laden des GPT-Modells von Hugging Face
4
- model = gr.Interface.load("Loewolf/GPT_1")
5
 
6
- # Erstellen der Gradio-Schnittstelle mit dem geladenen Modell
7
- interface = gr.Interface(fn=model, inputs="text", outputs="text")
 
 
 
 
8
 
9
  # Starten der Gradio-App
10
  interface.launch()
11
 
12
 
13
 
 
 
1
  import gradio as gr
2
+ from transformers import pipeline
3
 
4
+ # Laden des GPT-Modells mit Hugging Face Pipeline
5
+ model = pipeline("text-generation", model="Loewolf/GPT_1")
6
 
7
+ # Definition einer Wrapper-Funktion für das Modell
8
+ def generate_text(input_text):
9
+ return model(input_text)[0]['generated_text']
10
+
11
+ # Erstellen der Gradio-Schnittstelle
12
+ interface = gr.Interface(fn=generate_text, inputs="text", outputs="text")
13
 
14
  # Starten der Gradio-App
15
  interface.launch()
16
 
17
 
18
 
19
+