Erfan11 commited on
Commit
6060814
1 Parent(s): 7c4a3f6

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +11 -1
main.py CHANGED
@@ -25,4 +25,14 @@ def main():
25
  print(result)
26
 
27
  if __name__ == "__main__":
28
- main()
 
 
 
 
 
 
 
 
 
 
 
25
  print(result)
26
 
27
  if __name__ == "__main__":
28
+ main()
29
+ from transformers import BertForSequenceClassification
30
+
31
+ # Load the TensorFlow model using from_tf=True
32
+ model = BertForSequenceClassification.from_pretrained(
33
+ "Erfan11/Neuracraft",
34
+ from_tf=True,
35
+ use_auth_token="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"
36
+ )
37
+
38
+ # Additional code to run your app can go here (for example, Streamlit or Gradio interface)