HawkeyeHS commited on
Commit
9fdc2d8
·
1 Parent(s): 2ee266a

Fixed error

Browse files
Files changed (2) hide show
  1. app.py +7 -8
  2. requirements.txt +1 -0
app.py CHANGED
@@ -1,19 +1,13 @@
1
  import os
2
- import warnings
3
- from transformers import AutoModelForImageClassification, AutoFeatureExtractor
4
- import torch
5
  from flask_cors import CORS
6
- from flask import Flask, request, json, Response
7
- import numpy as np
8
- from PIL import Image
9
  import requests
10
- from io import BytesIO
11
  from bs4 import BeautifulSoup
12
  from urllib.parse import urljoin
13
  import google.generativeai as genai
14
  from langchain_google_genai import GoogleGenerativeAIEmbeddings
15
  from langchain_google_genai import ChatGoogleGenerativeAI
16
- import warnings
17
  from dotenv import load_dotenv
18
 
19
  os.environ["CUDA_VISIBLE_DEVICES"] = ""
@@ -62,6 +56,11 @@ def extract_images():
62
  api_key=os.getenv("GOOGLE_API_KEY")
63
  genai.configure(api_key=api_key)
64
  model=genai.GenerativeModel('gemini-pro')
 
 
 
 
 
65
 
66
  # @app.route("/predict", methods=["GET"])
67
  # def predict():
 
1
  import os
2
+ from transformers import pipeline
 
 
3
  from flask_cors import CORS
4
+ from flask import Flask, request, json
 
 
5
  import requests
 
6
  from bs4 import BeautifulSoup
7
  from urllib.parse import urljoin
8
  import google.generativeai as genai
9
  from langchain_google_genai import GoogleGenerativeAIEmbeddings
10
  from langchain_google_genai import ChatGoogleGenerativeAI
 
11
  from dotenv import load_dotenv
12
 
13
  os.environ["CUDA_VISIBLE_DEVICES"] = ""
 
56
  api_key=os.getenv("GOOGLE_API_KEY")
57
  genai.configure(api_key=api_key)
58
  model=genai.GenerativeModel('gemini-pro')
59
+ sentiment_analysis = pipeline("sentiment-analysis",model="siebert/sentiment-roberta-large-english")
60
+
61
+ # @app.route('/sentiment',methods=['POST'])
62
+ # def sentiment():
63
+
64
 
65
  # @app.route("/predict", methods=["GET"])
66
  # def predict():
requirements.txt CHANGED
@@ -139,6 +139,7 @@ torch==2.2.1
139
  torchmetrics==1.3.1
140
  torchvision==0.17.1
141
  tqdm==4.66.2
 
142
  typing-inspect==0.9.0
143
  typing_extensions==4.10.0
144
  tzdata==2024.1
 
139
  torchmetrics==1.3.1
140
  torchvision==0.17.1
141
  tqdm==4.66.2
142
+ transformers
143
  typing-inspect==0.9.0
144
  typing_extensions==4.10.0
145
  tzdata==2024.1