Canstralian commited on
Commit
5017c6b
·
verified ·
1 Parent(s): bec6e05

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -3,10 +3,17 @@ import streamlit as st
3
  from PyPDF2 import PdfReader
4
  from transformers import pipeline
5
 
6
- # Initialize the Hugging Face model pipeline
7
- model_name = "your-huggingface-model-name" # Replace with your model's name
 
 
8
  nlp_pipeline = pipeline("text2text-generation", model=model_name)
9
 
 
 
 
 
 
10
  def process_pdf(file):
11
  reader = PdfReader(file)
12
  text = ""
 
3
  from PyPDF2 import PdfReader
4
  from transformers import pipeline
5
 
6
+ # Specify the model name
7
+ model_name = "Canstralian/RabbitRedux"
8
+
9
+ # Initialize the pipeline
10
  nlp_pipeline = pipeline("text2text-generation", model=model_name)
11
 
12
+ # Example usage
13
+ input_text = "Provide an example of secure Python coding practices."
14
+ output = nlp_pipeline(input_text)
15
+ print(output)
16
+
17
  def process_pdf(file):
18
  reader = PdfReader(file)
19
  text = ""