Commit
•
6e6a199
1
Parent(s):
e3c04f6
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,8 +1,8 @@
|
|
| 1 |
from transformers import pipeline
|
| 2 |
import gradio as gr
|
| 3 |
|
| 4 |
-
# Initialize the pipeline
|
| 5 |
-
pipe = pipeline("text-classification", model="ZachBeesley/Spam-Detector")
|
| 6 |
|
| 7 |
# Function to process the input text and return the predicted label
|
| 8 |
def predict(text):
|
|
|
|
| 1 |
from transformers import pipeline
|
| 2 |
import gradio as gr
|
| 3 |
|
| 4 |
+
# Initialize the pipeline with TensorFlow
|
| 5 |
+
pipe = pipeline("text-classification", model="ZachBeesley/Spam-Detector", framework="tf")
|
| 6 |
|
| 7 |
# Function to process the input text and return the predicted label
|
| 8 |
def predict(text):
|